如何获取Spark MLlib线性回归(SGD)模型的输入列

时间:2017-05-28 17:07:48

标签: apache-spark machine-learning linear-regression apache-spark-mllib

我已经在Spark(MLlib)中成功构建了LinearRegressionModel。

我的要求是获取用于构建模型的输入列。有没有办法实现这个目标?

例如。考虑一下,我使用“ColA”,“ColB”和“ColC”来构建模型。在构建模型之后,有没有办法从模型对象中获取这些列名?

1 个答案:

答案 0 :(得分:0)

TL; DR 不,这是不可能的。

您应该使用PipelineTransformers来了解使用的内容和步骤。