我正在尝试从 Spark 中的逻辑回归模型中获取特征名称。
不知道该怎么做
这是我的代码
lr = LogisticRegression(labelCol="label", featuresCol="features", maxIter=10)
lrModel = lr.fit(trainingData)
lrPredictions = lrModel.transform(testData)
lrCv = CrossValidator(estimator=lr, estimatorParamMaps=lrParamGrid, evaluator=evaluator, numFolds=2)
lrCvModel = lrCv.fit(trainingData)
这就是我尝试过的
lr.summary._call_java('featuresCol')
#returns the error: 'LogisticRegression' object has no attribute 'summary'
lrCvModel.summary._java_obj.getFeaturesCol()
#returns error: AttributeError: 'CrossValidatorModel' object has no attribute 'summary'
您应该尝试访问最佳模型(文档(摘要:
lrCvModel.bestModel.summary
或逻辑回归模型摘要 (doc(
lrModel.summary