Class: ClassificationModel

eclairjs/mllib/classification. ClassificationModel

new ClassificationModel()

Represents a classification model that predicts to which of a set of categories an example belongs. The categories are represented by double values: 0.0, 1.0, 2.0, etc.
Source:

Methods

predict(testData) → {module:eclairjs.RDD|float}

Predict values for the given data set using the model trained.
Parameters:
Name Type Description
testData module:eclairjs.RDD | module:eclairjs/mllib/linalg.Vector RDD representing data points to be predicted or Vector array representing a single data point
Source:
Returns:
an RDD[float] where each entry contains the corresponding prediction or float predicted category from the trained model
Type
module:eclairjs.RDD | float