Package com.datarobot.prediction

Class BaseClassificationPredictor

java.lang.Object

com.datarobot.prediction.BasePredictor

com.datarobot.prediction.BaseClassificationPredictor

All Implemented Interfaces:

Serializable

public class BaseClassificationPredictor
extends BasePredictor
implements Serializable

See Also:

Serialized Form

Field Summary

Fields inherited from class com.datarobot.prediction.BasePredictor

features, numDoubles, numStrings, predictor

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	BaseClassificationPredictor(MulticlassPredictor predictor)
protected	BaseClassificationPredictor(Predictor predictor)

Method Summary

Modifier and Type	Method	Description
Map<String,Type>	getFeatures()	Returns an ordered map of the model's features and their types.
static BaseClassificationPredictor	getPredictor(String id)	Returns an instance of the loaded BaseClassificationPredictor by model ID.
Map<String,Double>	score(Iterable<? extends Object> row)	Predicts a single row.
Map<String,Double>	score(Map<String,? extends Object> row)	Predicts a single row.

Methods inherited from class com.datarobot.prediction.BasePredictor

extractFeatureValues, getModel

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

BaseClassificationPredictor

protected BaseClassificationPredictor(MulticlassPredictor predictor)

BaseClassificationPredictor

protected BaseClassificationPredictor(Predictor predictor)

Method Detail

score

public Map<String,Double> score(Iterable<? extends Object> row)
                         throws Exception

Predicts a single row.

The values of row should be in the same order as defined in BasePredictor.features

Parameters:

row - - The field values (String or Double)

Returns:

a mapping from a string class label to a double probability

Throws:

Exception

score

public Map<String,Double> score(Map<String,? extends Object> row)
                         throws Exception

Predicts a single row.

The row is a map of features to their respective values

Parameters:

row - - The field values (String or Double)

Returns:

a mapping from a string class label to a double probability

Throws:

Exception

getFeatures

public Map<String,Type> getFeatures()

Returns an ordered map of the model's features and their types. The order of fields matches the order expected by score(Iterable)

Returns:

an ordered map of the model's features.

getPredictor

public static BaseClassificationPredictor getPredictor(String id)
                                                throws ClassNotFoundException,
                                                       IllegalAccessException,
                                                       InstantiationException

Returns an instance of the loaded BaseClassificationPredictor by model ID.

Parameters:

id - - model id (name of the jar)

Returns:

BaseClassificationPredictor instance

Throws:

ClassNotFoundException - If model id was not found in classpath

InstantiationException - If model cannot be instantiated

IllegalAccessException - If model's major version is bigger than Predictor's major version