Interface IClassificationPredictor
-
- All Superinterfaces:
IPredictorInfo
public interface IClassificationPredictor extends IPredictorInfo
Interface for multi class and binary classification models.
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description String[]getClassLabels()Model specific class labels that will be predicted.Map<String,Double>score(Iterable<?> row)Scores a single row.Map<String,Double>score(Map<String,?> row)Scores a single row.-
Methods inherited from interface com.datarobot.prediction.compatible.IPredictorInfo
getFeatures, getModelId, getPredictorClass
-
-
-
-
Method Detail
-
getClassLabels
String[] getClassLabels()
Model specific class labels that will be predicted.NOTE:
These labels will be used as a key set for outputs of
score(Iterable)andscore(Map).- Returns:
- array of class labels.
-
score
Map<String,Double> score(Map<String,?> row)
Scores a single row.- Parameters:
row- row with feature values.- Returns:
- class label to its probability.
-
score
Map<String,Double> score(Iterable<?> row)
Scores a single row.NOTE:
Here
Iterable<?> rowis the feature values in the same order as they goes inIPredictorInfo.getFeatures().- Parameters:
row- row with feature values in right order.- Returns:
- class label to its probability.
-
-