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 Details

    • general

      public DataOptions general
      The data loading options.
    • centroids

      @Option(charName='n', longName="num-clusters", usage="Number of clusters to infer.") public int centroids
      Number of clusters to infer.
    • iterations

      @Option(charName='i', longName="iterations", usage="Maximum number of iterations.") public int iterations
      Maximum number of iterations.
    • distance

      @Option(charName='d', longName="distance", usage="Distance function to use in the e step.") public KMeansTrainer.Distance distance
      Distance function to use in the e step.
    • initialisation

      @Option(charName='s', longName="initialisation", usage="Type of initialisation to use for centroids.") public KMeansTrainer.Initialisation initialisation
      Type of initialisation to use for centroids.
    • numThreads

      @Option(charName='t', longName="num-threads", usage="Number of threads to use (range (1, num hw threads)).") public int numThreads
      Number of threads to use (range (1, num hw threads)).
  • Constructor Details

    • KMeansOptions

      public KMeansOptions()
  • Method Details

    • getOptionsDescription

      public String getOptionsDescription()
      Specified by:
      getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options