public class OneHotEncoderModel extends Model<OneHotEncoderModel> implements OneHotEncoderBase, MLWritable
Modifier and Type | Method and Description |
---|---|
int[] |
categorySizes() |
OneHotEncoderModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
BooleanParam |
dropLast()
Whether to drop the last category in the encoded vector (default: true)
|
Param<String> |
handleInvalid()
Param for how to handle invalid data during transform().
|
Param<String> |
inputCol()
Param for input column name.
|
StringArrayParam |
inputCols()
Param for input column names.
|
static OneHotEncoderModel |
load(String path) |
Param<String> |
outputCol()
Param for output column name.
|
StringArrayParam |
outputCols()
Param for output column names.
|
static MLReader<OneHotEncoderModel> |
read() |
OneHotEncoderModel |
setDropLast(boolean value) |
OneHotEncoderModel |
setHandleInvalid(String value) |
OneHotEncoderModel |
setInputCol(String value) |
OneHotEncoderModel |
setInputCols(String[] values) |
OneHotEncoderModel |
setOutputCol(String value) |
OneHotEncoderModel |
setOutputCols(String[] values) |
String |
toString() |
Dataset<Row> |
transform(Dataset<?> dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
Check transform validity and derive the output schema from the input schema.
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
MLWriter |
write()
Returns an
MLWriter instance for this ML instance. |
transform, transform, transform
params
getDropLast, getInOutCols, validateAndTransformSchema
getHandleInvalid
getInputCol
getInputCols
getOutputCol
getOutputCols
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
save
$init$, initializeForcefully, initializeLogIfNecessary, initializeLogIfNecessary, initializeLogIfNecessary$default$2, initLock, isTraceEnabled, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning, org$apache$spark$internal$Logging$$log__$eq, org$apache$spark$internal$Logging$$log_, uninitialize
public static MLReader<OneHotEncoderModel> read()
public static OneHotEncoderModel load(String path)
public Param<String> handleInvalid()
OneHotEncoderBase
handleInvalid
in interface OneHotEncoderBase
handleInvalid
in interface HasHandleInvalid
public final BooleanParam dropLast()
OneHotEncoderBase
dropLast
in interface OneHotEncoderBase
public final StringArrayParam outputCols()
HasOutputCols
outputCols
in interface HasOutputCols
public final Param<String> outputCol()
HasOutputCol
outputCol
in interface HasOutputCol
public final StringArrayParam inputCols()
HasInputCols
inputCols
in interface HasInputCols
public final Param<String> inputCol()
HasInputCol
inputCol
in interface HasInputCol
public String uid()
Identifiable
uid
in interface Identifiable
public int[] categorySizes()
public OneHotEncoderModel setInputCol(String value)
public OneHotEncoderModel setOutputCol(String value)
public OneHotEncoderModel setInputCols(String[] values)
public OneHotEncoderModel setOutputCols(String[] values)
public OneHotEncoderModel setDropLast(boolean value)
public OneHotEncoderModel setHandleInvalid(String value)
public StructType transformSchema(StructType schema)
PipelineStage
We check validity for interactions between parameters during transformSchema
and
raise an exception if any parameter value is invalid. Parameter value checks which
do not depend on other parameters are handled by Param.validate()
.
Typical implementation should first conduct verification on schema change and parameter validity, including complex parameter interaction checks.
transformSchema
in class PipelineStage
schema
- (undocumented)public Dataset<Row> transform(Dataset<?> dataset)
Transformer
transform
in class Transformer
dataset
- (undocumented)public OneHotEncoderModel copy(ParamMap extra)
Params
defaultCopy()
.copy
in interface Params
copy
in class Model<OneHotEncoderModel>
extra
- (undocumented)public MLWriter write()
MLWritable
MLWriter
instance for this ML instance.write
in interface MLWritable
public String toString()
toString
in interface Identifiable
toString
in class Object