org.apache.spark.ml.classification

Class BinaryLogisticRegressionTrainingSummaryImpl

Object

org.apache.spark.ml.classification.LogisticRegressionSummaryImpl

org.apache.spark.ml.classification.BinaryLogisticRegressionSummaryImpl

org.apache.spark.ml.classification.BinaryLogisticRegressionTrainingSummaryImpl

All Implemented Interfaces:

java.io.Serializable, BinaryClassificationSummary, BinaryLogisticRegressionSummary, BinaryLogisticRegressionTrainingSummary, ClassificationSummary, LogisticRegressionSummary, LogisticRegressionTrainingSummary, TrainingSummary

public class BinaryLogisticRegressionTrainingSummaryImpl
extends BinaryLogisticRegressionSummaryImpl
implements BinaryLogisticRegressionTrainingSummary

Binary logistic regression training results.

param: predictions dataframe output by the model's transform method. param: probabilityCol field in "predictions" which gives the probability of each class as a vector. 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: featuresCol field in "predictions" which gives the features of each instance as a vector. 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
BinaryLogisticRegressionTrainingSummaryImpl(Dataset<Row> predictions, String probabilityCol, String predictionCol, String labelCol, String featuresCol, 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.BinaryLogisticRegressionSummaryImpl

areaUnderROC, fMeasureByThreshold, pr, precisionByThreshold, recallByThreshold, roc

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

featuresCol, labelCol, predictionCol, predictions, probabilityCol, weightCol

Methods inherited from class Object

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

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

scoreCol

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

areaUnderROC, fMeasureByThreshold, pr, precisionByThreshold, recallByThreshold, roc

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

asBinary, featuresCol, probabilityCol

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

BinaryLogisticRegressionTrainingSummaryImpl

public BinaryLogisticRegressionTrainingSummaryImpl(Dataset<Row> predictions,
                                                   String probabilityCol,
                                                   String predictionCol,
                                                   String labelCol,
                                                   String featuresCol,
                                                   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)