DaalNaiveBayesModel train


train(self, frame, label_column, observation_columns, num_classes=2)

[BETA] Build a multinomial naive bayes model.

Parameters:

frame : Frame

A frame to train the model on.

label_column : unicode

Column containing the label for each observation.

observation_columns : list

Column(s) containing the observations.

num_classes : int32 (default=2)

Number of classes

Returns:

: dict

dictionary A dictionary with trained multinomial naive bayes model with the following keys: ‘class_log_prior’: smoothed empirical log probability for each class ‘feature_log_prob’: empirical log probability of features given a class, P(x_i|y)

Train a DaalNaiveBayesModel using the observation column, label column of the train frame and an optional lambda value.

Examples

See here for examples.