Class BinaryRandomForestClassificationTrainingSummaryImpl

All Implemented Interfaces:
Serializable, BinaryClassificationSummary, BinaryRandomForestClassificationSummary, BinaryRandomForestClassificationTrainingSummary, ClassificationSummary, RandomForestClassificationSummary, RandomForestClassificationTrainingSummary, TrainingSummary

public class BinaryRandomForestClassificationTrainingSummaryImpl extends BinaryRandomForestClassificationSummaryImpl implements BinaryRandomForestClassificationTrainingSummary
Binary RandomForestClassification training results.

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

See Also:
  • Constructor Details

    • BinaryRandomForestClassificationTrainingSummaryImpl

      public BinaryRandomForestClassificationTrainingSummaryImpl(Dataset<Row> predictions, String scoreCol, String predictionCol, String labelCol, String weightCol, double[] objectiveHistory)
  • Method Details

    • 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)