org.tribuo.classification.mnb

Class MultinomialNaiveBayesOptions

java.lang.Object

org.tribuo.classification.mnb.MultinomialNaiveBayesOptions

All Implemented Interfaces:

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

public class MultinomialNaiveBayesOptions
extends Object
implements ClassificationOptions<MultinomialNaiveBayesTrainer>

CLI options for a multinomial naive bayes model.

Field Summary

Fields

Modifier and Type	Field and Description
double	mnbAlpha

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

header

Constructor Summary

Constructors

Constructor and Description
MultinomialNaiveBayesOptions()

Method Summary

Modifier and Type	Method and Description
MultinomialNaiveBayesTrainer	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, getEnumConstantString, getOptionFields, getOptions, getOptionsDescription, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

mnbAlpha

@Option(charName=97,
        longName="mnb-alpha",
        usage="Smoothing parameter for the conditional probabilities.")
public double mnbAlpha

Constructor Detail

MultinomialNaiveBayesOptions

public MultinomialNaiveBayesOptions()

Method Detail

getTrainer

public MultinomialNaiveBayesTrainer getTrainer()

Description copied from interface: ClassificationOptions

Constructs the trainer based on the provided arguments.

Specified by:

getTrainer in interface ClassificationOptions<MultinomialNaiveBayesTrainer>

Returns:

The trainer.