Dominik Jastrzębski created SPARK-14844:
-------------------------------------------
Summary: KMeansModel in spark.ml should allow to change featureCol
and predictionCol
Key: SPARK-14844
URL: https://issues.apache.org/jira/browse/SPARK-14844
Project: Spark
Issue Type: Improvement
Components: ML
Affects Versions: 1.6.1, 1.6.0
Reporter: Dominik Jastrzębski
We need to add setFeaturesCol, setPredictionCol methods in
org.apache.spark.ml.clustering.KMeansModel.
This will allow to:
* transform a DataFrame with different feature column name than in the
DataFrame the model was fitted on.
* create a prediction column with name other than the name that was set during
model fitting.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
