Package org.apache.spark.ml.classification

Class RandomForestClassificationSummaryImpl

Object

org.apache.spark.ml.classification.RandomForestClassificationSummaryImpl

All Implemented Interfaces:

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	Description
RandomForestClassificationSummaryImpl(Dataset<Row> predictions, String predictionCol, String labelCol, String weightCol)

Method Summary

Modifier and Type	Method	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 java.lang.Object

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

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

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

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

asBinary

Constructor Details

RandomForestClassificationSummaryImpl

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

Method Details

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