org.apache.spark.ml.param.shared

Interface HasThresholds

All Superinterfaces:

Identifiable, Params, java.io.Serializable

All Known Subinterfaces:

DecisionTreeClassifierParams, FMClassifierParams, GBTClassifierParams, LogisticRegressionParams, MultilayerPerceptronParams, ProbabilisticClassifierParams, RandomForestClassifierParams, TreeEnsembleClassifierParams

All Known Implementing Classes:

Binarizer, DecisionTreeClassificationModel, DecisionTreeClassifier, FMClassificationModel, FMClassifier, GBTClassificationModel, GBTClassifier, LogisticRegression, LogisticRegressionModel, MultilayerPerceptronClassificationModel, MultilayerPerceptronClassifier, NaiveBayes, NaiveBayesModel, ProbabilisticClassificationModel, ProbabilisticClassifier, RandomForestClassificationModel, RandomForestClassifier

public interface HasThresholds
extends Params

Trait for shared param thresholds. This trait may be changed or removed between minor versions.

Method Summary

Modifier and Type	Method and Description
double[]	getThresholds()
DoubleArrayParam	thresholds() Param for Thresholds in multi-class classification to adjust the probability of predicting each class.

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Method Detail

getThresholds

double[] getThresholds()

thresholds

DoubleArrayParam thresholds()

Param for Thresholds in multi-class classification to adjust the probability of predicting each class. Array must have length equal to the number of classes, with values > 0 excepting that at most one value may be 0. The class with largest value p/t is predicted, where p is the original probability of that class and t is the class's threshold.

Returns:

(undocumented)