public final class VectorSlicer extends Transformer
The subset of features can be specified with either indices (setIndices())
or names (setNames()). At least one feature must be selected. Duplicate features
are not allowed, so there can be no overlap between selected indices and names.
The output vector will order features with the selected indices first (in the order given), followed by the selected names (in the order given).
| Constructor and Description |
|---|
VectorSlicer() |
VectorSlicer(String uid) |
| Modifier and Type | Method and Description |
|---|---|
VectorSlicer |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
int[] |
getIndices() |
String[] |
getNames() |
IntArrayParam |
indices()
An array of indices to select features from a vector column.
|
static VectorSlicer |
load(String path) |
StringArrayParam |
names()
An array of feature names to select features from a vector column.
|
VectorSlicer |
setIndices(int[] value) |
VectorSlicer |
setInputCol(String value) |
VectorSlicer |
setNames(String[] value) |
VectorSlicer |
setOutputCol(String value) |
DataFrame |
transform(DataFrame dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
void |
validateParams() |
transform, transform, transformequals, 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, shouldOwntoStringinitializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarningpublic static VectorSlicer load(String path)
public String uid()
Identifiablepublic IntArrayParam indices()
names.
Default: Empty arraypublic int[] getIndices()
public VectorSlicer setIndices(int[] value)
public StringArrayParam names()
Attributes.
There can be no overlap with indices.
Default: Empty Arraypublic String[] getNames()
public VectorSlicer setNames(String[] value)
public VectorSlicer setInputCol(String value)
public VectorSlicer setOutputCol(String value)
public void validateParams()
public DataFrame transform(DataFrame dataset)
Transformertransform in class Transformerdataset - (undocumented)public StructType transformSchema(StructType schema)
PipelineStageDerives the output schema from the input schema.
transformSchema in class PipelineStageschema - (undocumented)public VectorSlicer copy(ParamMap extra)
Paramscopy in interface Paramscopy in class Transformerextra - (undocumented)defaultCopy()