Class ClassificationEnsembleOptions

All Implemented Interfaces:

public class ClassificationEnsembleOptions extends Object implements
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
      Ensemble method, options are {ADABOOST, BAGGING, RF}.
    • ensembleSize

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

      @Option(longName="ensemble-seed", usage="RNG seed.") public long seed
      RNG 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.
      trainer - The trainer to wrap.
      An ensemble trainer.