public final class OneVsRestModel extends Model<OneVsRestModel>
OneVsRest.
This stores the models resulting from training k binary classifiers: one for each class.
Each example is scored against all k models, and the model with the highest score
is picked to label the example.
param: labelMetadata Metadata of label column if it exists, or Nominal attribute representing the number of classes in training dataset otherwise. param: models The binary classification models for the reduction. The i-th model is produced by testing the i-th class (taking label 1) vs the rest (taking label 0).
| Modifier and Type | Method and Description |
|---|---|
Param<Classifier<?,? extends Classifier<java.lang.Object,Classifier,ClassificationModel>,? extends ClassificationModel<java.lang.Object,ClassificationModel>>> |
classifier()
param for the base binary classifier that we reduce multiclass classification into.
|
OneVsRestModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
Classifier<?,? extends Classifier<java.lang.Object,Classifier,ClassificationModel>,? extends ClassificationModel<java.lang.Object,ClassificationModel>> |
getClassifier() |
java.lang.Object |
models() |
DataFrame |
transform(DataFrame dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
java.lang.String |
uid()
An immutable unique ID for the object and its derivatives.
|
transform, transform, transformtransformSchemaclone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitclear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn, validateParamstoStringinitializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarningpublic java.lang.String uid()
Identifiableuid in interface Identifiablepublic java.lang.Object models()
public StructType transformSchema(StructType schema)
PipelineStageDerives the output schema from the input schema.
transformSchema in class PipelineStageschema - (undocumented)public DataFrame transform(DataFrame dataset)
Transformertransform in class Transformerdataset - (undocumented)public OneVsRestModel copy(ParamMap extra)
Paramscopy in interface Paramscopy in class Model<OneVsRestModel>extra - (undocumented)defaultCopy()public Param<Classifier<?,? extends Classifier<java.lang.Object,Classifier,ClassificationModel>,? extends ClassificationModel<java.lang.Object,ClassificationModel>>> classifier()
OneVsRest.public Classifier<?,? extends Classifier<java.lang.Object,Classifier,ClassificationModel>,? extends ClassificationModel<java.lang.Object,ClassificationModel>> getClassifier()