Class KMeansOptions

java.lang.Object
org.tribuo.clustering.kmeans.KMeansOptions
All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options

public class KMeansOptions extends Object implements com.oracle.labs.mlrg.olcut.config.Options
OLCUT Options for the K-Means implementation.
  • Field Details

    • iterations

      @Option(longName="kmeans-interations", usage="Iterations of the k-means algorithm. Defaults to 10.") public int iterations
      Iterations of the k-means algorithm. Defaults to 10.
    • centroids

      @Option(longName="kmeans-num-centroids", usage="Number of centroids in K-Means. Defaults to 10.") public int centroids
      Number of centroids in K-Means. Defaults to 10.
    • distance

      @Option(longName="kmeans-distance", usage="Distance function in K-Means. Defaults to EUCLIDEAN.") public KMeansTrainer.Distance distance
      Distance function in K-Means. Defaults to EUCLIDEAN.
    • initialisation

      @Option(longName="kmeans-initialisation", usage="Initialisation function in K-Means. Defaults to RANDOM.") public KMeansTrainer.Initialisation initialisation
      Initialisation function in K-Means. Defaults to RANDOM.
    • numThreads

      @Option(longName="kmeans-num-threads", usage="Number of computation threads in K-Means. Defaults to 4.") public int numThreads
      Number of computation threads in K-Means. Defaults to 4.
  • Constructor Details

    • KMeansOptions

      public KMeansOptions()
  • Method Details

    • getTrainer

      public KMeansTrainer getTrainer()
      Gets the configured KMeansTrainer using the options in this object.
      Returns:
      A KMeansTrainer.