diff --git a/Classification/Core/src/main/java/org/tribuo/classification/ensemble/ClassificationEnsembleOptions.java b/Classification/Core/src/main/java/org/tribuo/classification/ensemble/ClassificationEnsembleOptions.java index 78539451a..a4ba507db 100644 --- a/Classification/Core/src/main/java/org/tribuo/classification/ensemble/ClassificationEnsembleOptions.java +++ b/Classification/Core/src/main/java/org/tribuo/classification/ensemble/ClassificationEnsembleOptions.java @@ -55,15 +55,25 @@ public enum EnsembleType { RF } - @Option(longName="ensemble-type",usage="Ensemble method, options are {ADABOOST, BAGGING, RF}.") + /** + * Ensemble method, options are {ADABOOST, BAGGING, RF}. + */ + @Option(longName = "ensemble-type", usage = "Ensemble method, options are {ADABOOST, BAGGING, RF}.") public EnsembleType type = EnsembleType.BAGGING; - @Option(longName="ensemble-size",usage="Number of base learners in the ensemble.") + /** + * Number of base learners in the ensemble. + */ + @Option(longName = "ensemble-size", usage = "Number of base learners in the ensemble.") public int ensembleSize = -1; - @Option(longName="ensemble-seed",usage="RNG seed.") + /** + * RNG seed. + */ + @Option(longName = "ensemble-seed", usage = "RNG seed.") public long seed = Trainer.DEFAULT_SEED; /** * Wraps the supplied trainer using the ensemble trainer described by these options. + * * @param trainer The trainer to wrap. * @return An ensemble trainer. */ @@ -72,21 +82,21 @@ public Trainer