Package 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	Description
double	mnbAlpha	Smoothing parameter for the conditional probabilities.

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

header

Constructor Summary

Constructors

Constructor	Description
MultinomialNaiveBayesOptions()

Method Summary

Modifier and Type	Method	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

getOptionsDescription

Field Details

mnbAlpha

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

Smoothing parameter for the conditional probabilities.

Constructor Details

MultinomialNaiveBayesOptions

public MultinomialNaiveBayesOptions()

Method Details

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.