Class ClassificationEnsembleOptions

java.lang.Object
org.tribuo.classification.ensemble.ClassificationEnsembleOptions
All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
public class ClassificationEnsembleOptions extends Object implements com.oracle.labs.mlrg.olcut.config.Options
Options for building a classification ensemble.

  • Field Details

    • type

      @Option(longName="ensemble-type", usage="Ensemble method, options are {ADABOOST, BAGGING, RF}.") public ClassificationEnsembleOptions.EnsembleType type

    • ensembleSize

      @Option(longName="ensemble-size", usage="Number of base learners in the ensemble.") public int ensembleSize

    • seed

      @Option(longName="ensemble-seed", usage="RNG seed.") public long seed

  • Constructor Details

    • ClassificationEnsembleOptions

      public ClassificationEnsembleOptions()

  • Method Details

    • wrapTrainer

      public Trainer<Label> wrapTrainer(Trainer<Label> trainer)
      Wraps the supplied trainer using the ensemble trainer described by these options.
      Parameters:
      trainer - The trainer to wrap.
      Returns:
      An ensemble trainer.