Package org.apache.spark.ml.param.shared

Interface HasProbabilityCol

All Superinterfaces:
Identifiable, Params, Serializable, scala.Serializable
All Known Subinterfaces:
DecisionTreeClassifierParams, FMClassifierParams, GaussianMixtureParams, GBTClassifierParams, LogisticRegressionParams, MultilayerPerceptronParams, ProbabilisticClassifierParams, RandomForestClassifierParams, TreeEnsembleClassifierParams
All Known Implementing Classes:
DecisionTreeClassificationModel, DecisionTreeClassifier, FMClassificationModel, FMClassifier, GaussianMixture, GaussianMixtureModel, GBTClassificationModel, GBTClassifier, LogisticRegression, LogisticRegressionModel, MulticlassClassificationEvaluator, MultilayerPerceptronClassificationModel, MultilayerPerceptronClassifier, NaiveBayes, NaiveBayesModel, ProbabilisticClassificationModel, ProbabilisticClassifier, RandomForestClassificationModel, RandomForestClassifier
public interface HasProbabilityCol extends Params
Trait for shared param probabilityCol (default: "probability"). This trait may be changed or removed between minor versions.

  • Method Details

    • getProbabilityCol

      String getProbabilityCol()

    • probabilityCol

      Param<String> probabilityCol()
      Param for Column name for predicted class conditional probabilities. Note: Not all models output well-calibrated probability estimates! These probabilities should be treated as confidences, not precise probabilities.
      Returns:
      (undocumented)