[我以前在Spark 2.2.0中使用过Logistic回归模型,并且通过对某些带标签的测试数据调用模型的evaluate()
方法,我能够获得BinaryLogisticRegressionSummary
对象,然后将其用于一些下游任务。
现在我正在使用PipelineModel
,其中最后一个阶段是逻辑回归模型。问题是PipelineModel
没有evaluate
方法,而只有transform
方法,该方法输出包含预测和标签的数据帧。是否有任何方法可以从BinaryLogisticRegressionSummary
的变换方法的输出中获取PipelineModel
对象,或以其他任何方式使用PipelineModel
?
lr = thePipelineModel.stages[-1]
的东西来获取对(您将期望的)正确对象的引用,然后对其进行评估。