org.tribuo.common.nearest

Class KNNClassifierOptions

java.lang.Object

org.tribuo.common.nearest.KNNClassifierOptions

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.config.Options, ClassificationOptions<KNNTrainer<Label>>

public class KNNClassifierOptions
extends Object
implements ClassificationOptions<KNNTrainer<Label>>

CLI Options for training a k-nearest neighbour predictor.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	KNNClassifierOptions.EnsembleCombinerType The type of combination function.

Field Summary

Fields

Modifier and Type	Field and Description
KNNModel.Backend	knnBackend
KNNTrainer.Distance	knnDistance
KNNClassifierOptions.EnsembleCombinerType	knnEnsembleCombiner
int	knnK
int	knnNumThreads

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

header

Constructor Summary

Constructors

Constructor and Description
KNNClassifierOptions()

Method Summary

Modifier and Type	Method and Description
String	getOptionsDescription()
KNNTrainer<Label>	getTrainer() Constructs the trainer based on the provided arguments.

Methods inherited from class java.lang.Object

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

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

formatUsage, generateTypeDescription, getAllOptions, getOptionFields, getOptions, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

knnK

@Option(longName="knn-k",
        usage="K nearest neighbours to use. Defaults to 1.")
public int knnK

knnNumThreads

@Option(longName="knn-num-threads",
        usage="Number of threads to use. Defaults to 1.")
public int knnNumThreads

knnDistance

@Option(longName="knn-distance",
        usage="Distance metric to use. Defaults to L2.")
public KNNTrainer.Distance knnDistance

knnBackend

@Option(longName="knn-backend",
        usage="Parallel backend to use.")
public KNNModel.Backend knnBackend

knnEnsembleCombiner

@Option(longName="knn-voting",
        usage="Parallel backend to use.")
public KNNClassifierOptions.EnsembleCombinerType knnEnsembleCombiner

Constructor Detail

KNNClassifierOptions

public KNNClassifierOptions()

Method Detail

getOptionsDescription

public String getOptionsDescription()

Specified by:

getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options

getTrainer

public KNNTrainer<Label> getTrainer()

Description copied from interface: ClassificationOptions

Constructs the trainer based on the provided arguments.

Specified by:

getTrainer in interface ClassificationOptions<KNNTrainer<Label>>

Returns:

The trainer.