smile.classification

Interface DataFrameClassifier

All Known Implementing Classes:

AdaBoost, DecisionTree, GradientTreeBoost, RandomForest

public interface DataFrameClassifier

Classification trait on DataFrame.

Method Summary

Modifier and Type	Method and Description
smile.data.formula.Formula	formula() Returns the formula associated with the model.
default int[]	predict(smile.data.DataFrame data) Predicts the class labels of a data frame.
int	predict(smile.data.Tuple x) Predicts the class label of an instance.
smile.data.type.StructType	schema() Returns the design matrix schema.

Method Detail

predict

int predict(smile.data.Tuple x)

Predicts the class label of an instance.

Parameters:

x - a tuple instance.

Returns:

the predicted class label.

predict

default int[] predict(smile.data.DataFrame data)

Predicts the class labels of a data frame.

Parameters:

data - the data frame.

Returns:

the predicted class labels.

formula

smile.data.formula.Formula formula()

Returns the formula associated with the model.

schema

smile.data.type.StructType schema()

Returns the design matrix schema.