Spark: Extracting summary for a ML logistic regression model from a pipeline model

前端 未结 1 481
你的背包
你的背包 2021-01-19 05:19

I\'ve estimated a logistic regression using pipelines.

My last few lines before fitting the logistic regression:

from pyspark.ml.feature import Vect         


        
1条回答
  •  傲寒
    傲寒 (楼主)
    2021-01-19 06:01

    Just get the model from stages:

    lrModel.stages[-1].summary
    

    If model is earlier in the Pipeline replace -1 with its index.

    0 讨论(0)
提交回复
热议问题