org.apache.spark.ml.classification

Class RandomForestClassificationSummaryImpl

Object

org.apache.spark.ml.classification.RandomForestClassificationSummaryImpl

All Implemented Interfaces:

java.io.Serializable, ClassificationSummary, RandomForestClassificationSummary

Direct Known Subclasses:

BinaryRandomForestClassificationSummaryImpl, RandomForestClassificationTrainingSummaryImpl

public class RandomForestClassificationSummaryImpl
extends Object
implements RandomForestClassificationSummary

Multiclass RandomForestClassification results for a given model.

param: predictions dataframe output by the model's transform method. param: predictionCol field in "predictions" which gives the prediction for a data instance as a double. param: labelCol field in "predictions" which gives the true label of each instance. param: weightCol field in "predictions" which gives the weight of each instance.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
RandomForestClassificationSummaryImpl(Dataset<Row> predictions, String predictionCol, String labelCol, String weightCol)

Method Summary

Modifier and Type	Method and Description
String	labelCol() Field in "predictions" which gives the true label of each instance (if available).
String	predictionCol() Field in "predictions" which gives the prediction of each class.
Dataset<Row>	predictions() Dataframe output by the model's transform method.
String	weightCol() Field in "predictions" which gives the weight of each instance.

Methods inherited from class Object

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

Methods inherited from interface org.apache.spark.ml.classification.RandomForestClassificationSummary

asBinary

Methods inherited from interface org.apache.spark.ml.classification.ClassificationSummary

accuracy, falsePositiveRateByLabel, fMeasureByLabel, fMeasureByLabel, labels, precisionByLabel, recallByLabel, truePositiveRateByLabel, weightedFalsePositiveRate, weightedFMeasure, weightedFMeasure, weightedPrecision, weightedRecall, weightedTruePositiveRate

Constructor Detail

RandomForestClassificationSummaryImpl

public RandomForestClassificationSummaryImpl(Dataset<Row> predictions,
                                             String predictionCol,
                                             String labelCol,
                                             String weightCol)

Method Detail

labelCol

public String labelCol()

Description copied from interface: ClassificationSummary

Field in "predictions" which gives the true label of each instance (if available).

Specified by:

labelCol in interface ClassificationSummary

predictionCol

public String predictionCol()

Description copied from interface: ClassificationSummary

Field in "predictions" which gives the prediction of each class.

Specified by:

predictionCol in interface ClassificationSummary

predictions

public Dataset<Row> predictions()

Description copied from interface: ClassificationSummary

Dataframe output by the model's transform method.

Specified by:

predictions in interface ClassificationSummary

Returns:

(undocumented)

weightCol

public String weightCol()

Description copied from interface: ClassificationSummary

Field in "predictions" which gives the weight of each instance.

Specified by:

weightCol in interface ClassificationSummary