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

Question

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

我的要求是获取用于构建模型的输入列。有没有可能的方法来实现这一点？

例如。考虑一下，我使用“ColA”、“ColB”和“ColC”来构建模型。一旦我构建了模型，有没有办法从模型对象中获取这些列名？

Answer 1

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

您应该使用您的Pipeline 和Transformers 来了解使用了什么以及在哪一步。