NaiveBayesModel train

---

train(self, frame, label_column, observation_columns, lambda_parameter=None)

[ALPHA] Train a Naive Bayes model.

Parameters:

frame : <bound method AtkEntityType.__name__ of <trustedanalytics.rest.jsonschema.AtkEntityType object at 0x7f9e686f3fd0>>

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.

lambda_parameter : float64 (default=None)

Additive smoothing parameter Default is 1.0.

Returns:

: _Unit

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

Examples

>>> my_model = ta.NaiveBayesModel(name='naivebayesmodel')
>>> my_model.train(train_frame, 'name_of_label_column',['name_of_observation_column(s)'],0.9)