从Spark 2.2.0中的管道模型输出中获取BinaryLogisticRegressionSummary

问题描述 投票:0回答:1

[我以前在Spark 2.2.0中使用过Logistic回归模型,并且通过对某些带标签的测试数据调用模型的evaluate()方法,我能够获得BinaryLogisticRegressionSummary对象,然后将其用于一些下游任务。

现在我正在使用PipelineModel,其中最后一个阶段是逻辑回归模型。问题是PipelineModel没有evaluate方法,而只有transform方法,该方法输出包含预测和标签的数据帧。是否有任何方法可以从BinaryLogisticRegressionSummary的变换方法的输出中获取PipelineModel对象,或以其他任何方式使用PipelineModel

apache-spark machine-learning logistic-regression apache-spark-mllib apache-spark-ml
1个回答
0
投票
由于您知道PipelineModel的最后阶段是您需要评估的LogisticRegressionModel,请尝试使用类似lr = thePipelineModel.stages[-1]的东西来获取对(您将期望的)正确对象的引用,然后对其进行评估。
© www.soinside.com 2019 - 2024. All rights reserved.