trait ClassificationModel extends Serializable
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.
- Annotations
- @Since( "0.8.0" )
- Source
- ClassificationModel.scala
- Alphabetic
- By Inheritance
- ClassificationModel
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Abstract Value Members
-
abstract
def
predict(testData: Vector): Double
Predict values for a single data point using the model trained.
Predict values for a single data point using the model trained.
- testData
array representing a single data point
- returns
predicted category from the trained model
- Annotations
- @Since( "1.0.0" )
-
abstract
def
predict(testData: RDD[Vector]): RDD[Double]
Predict values for the given data set using the model trained.
Predict values for the given data set using the model trained.
- testData
RDD representing data points to be predicted
- returns
an RDD[Double] where each entry contains the corresponding prediction
- Annotations
- @Since( "1.0.0" )
Concrete Value Members
-
def
predict(testData: JavaRDD[Vector]): JavaRDD[Double]
Predict values for examples stored in a JavaRDD.
Predict values for examples stored in a JavaRDD.
- testData
JavaRDD representing data points to be predicted
- returns
a JavaRDD[java.lang.Double] where each entry contains the corresponding prediction
- Annotations
- @Since( "1.0.0" )