org.apache.spark.ml.classification

Class FMClassificationTrainingSummaryImpl

Object

org.apache.spark.ml.classification.FMClassificationSummaryImpl

org.apache.spark.ml.classification.FMClassificationTrainingSummaryImpl

All Implemented Interfaces:

java.io.Serializable, BinaryClassificationSummary, ClassificationSummary, FMClassificationSummary, FMClassificationTrainingSummary, TrainingSummary

public class FMClassificationTrainingSummaryImpl
extends FMClassificationSummaryImpl
implements FMClassificationTrainingSummary

FMClassifier training results.

param: predictions dataframe output by the model's transform method. param: scoreCol field in "predictions" which gives the probability of each instance. 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. param: objectiveHistory objective function (scaled loss + regularization) at each iteration.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
FMClassificationTrainingSummaryImpl(Dataset<Row> predictions, String scoreCol, String predictionCol, String labelCol, String weightCol, double[] objectiveHistory)

Method Summary

Modifier and Type	Method and Description
double[]	objectiveHistory() objective function (scaled loss + regularization) at each iteration.

Methods inherited from class org.apache.spark.ml.classification.FMClassificationSummaryImpl

areaUnderROC, fMeasureByThreshold, labelCol, pr, precisionByThreshold, predictionCol, predictions, recallByThreshold, roc, scoreCol, weightCol

Methods inherited from class Object

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

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

areaUnderROC, fMeasureByThreshold, pr, precisionByThreshold, recallByThreshold, roc, scoreCol

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

accuracy, falsePositiveRateByLabel, fMeasureByLabel, fMeasureByLabel, labelCol, labels, precisionByLabel, predictionCol, predictions, recallByLabel, truePositiveRateByLabel, weightCol, weightedFalsePositiveRate, weightedFMeasure, weightedFMeasure, weightedPrecision, weightedRecall, weightedTruePositiveRate

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

totalIterations

Constructor Detail

FMClassificationTrainingSummaryImpl

public FMClassificationTrainingSummaryImpl(Dataset<Row> predictions,
                                           String scoreCol,
                                           String predictionCol,
                                           String labelCol,
                                           String weightCol,
                                           double[] objectiveHistory)

Method Detail

objectiveHistory

public double[] objectiveHistory()

Description copied from interface: TrainingSummary

objective function (scaled loss + regularization) at each iteration. It contains one more element, the initial state, than number of iterations.

Specified by:

objectiveHistory in interface TrainingSummary

Returns:

(undocumented)