Class TrainTest.KMeansOptions
java.lang.Object
org.tribuo.clustering.kmeans.TrainTest.KMeansOptions
- All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
- Enclosing class:
TrainTest
public static class TrainTest.KMeansOptions
extends Object
implements com.oracle.labs.mlrg.olcut.config.Options
Options for the K-Means CLI.
-
Field Summary
FieldsModifier and TypeFieldDescriptionint
int
int
Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options
header
-
Constructor Summary
Constructors -
Method Summary
-
Field Details
-
general
-
centroids
@Option(charName='n', longName="num-clusters", usage="Number of clusters to infer.") public int centroids -
iterations
@Option(charName='i', longName="iterations", usage="Maximum number of iterations.") public int iterations -
distance
@Option(charName='d', longName="distance", usage="Distance function to use in the e step.") public KMeansTrainer.Distance distance -
initialisation
@Option(charName='s', longName="initialisation", usage="Type of initialisation to use for centroids.") public KMeansTrainer.Initialisation initialisation -
numThreads
@Option(charName='t', longName="num-threads", usage="Number of threads to use (range (1, num hw threads)).") public int numThreads
-
-
Constructor Details
-
KMeansOptions
public KMeansOptions()
-
-
Method Details
-
getOptionsDescription
- Specified by:
getOptionsDescription
in interfacecom.oracle.labs.mlrg.olcut.config.Options
-