关于的模型训练结果

Question

我是 scala ML包的新手。

在组装了一个管道并将一些回归模型拟合到训练数据集(使用命令：val model = pipeline.fit(training))之后，我如何检查/打印出拟合的模型细节:系数、统计数据、R2等等？

类似于R语言中的"summary(model)“？

谢谢!

Answer 1

您可以做的是，得到适当的阶段，然后查看其内容：

val pipelineModel: PipelineModel = pipelineData.fit(data)
val transformedColumns = pipelineModel.stages
val lda = transformedColumns(9).asInstanceOf[LDAModel]
// 9 is the index of the stage in the pipeline. If you just have a single stage, it would be zero then

在上述情况下，一旦您拥有了LDAModel，现在就可以查看统计数据了。

LDAModel只是一个例子。如果您的管道阶段是CountVectorizer，那么就使用CountVectorizerModel。如果您使用IDF，那么使用IDFModel。