[ https://issues.apache.org/jira/browse/SPARK-5895?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14386365#comment-14386365
]
Xusen Yin commented on SPARK-5895:
----------------------------------
I have another concern here.
We can not reveal each column name of a `Vector`. Given the selected features "age" and "salary",
how to select these two columns from a vector? One solution is giving it a list of column
names, say, `setColumnNames(List[String])`. But a more natural way to solve it is adding the
list of column names in `VectorAssembler`.
> Add VectorSlicer
> ----------------
>
> Key: SPARK-5895
> URL: https://issues.apache.org/jira/browse/SPARK-5895
> Project: Spark
> Issue Type: Sub-task
> Components: ML
> Reporter: Xiangrui Meng
>
> `VectorSlicer` takes a vector column and output a vector column with a subset of features.
> {code}
> val vs = new VectorSlicer()
> .setInputCol("user")
> .setSelectedFeatures("age", "salary")
> .setOutputCol("usefulUserFeatures")
> {code}
> We should allow specifying selected features by indices and by names. It should preserve
the output names.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org
|