Package org.tribuo.clustering.kmeans

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

Fields

Modifier and Type	Field	Description
int	centroids	Number of clusters to infer.
KMeansTrainer.Distance	distance	Distance function to use in the e step.
DataOptions	general	The data loading options.
KMeansTrainer.Initialisation	initialisation	Type of initialisation to use for centroids.
int	iterations	Maximum number of iterations.
int	numThreads	Number of threads to use (range (1, num hw threads)).

Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options

header

Constructor Summary

Constructors

Constructor	Description
KMeansOptions()

Method Summary

Modifier and Type	Method	Description
String	getOptionsDescription()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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