在pyspark中使用逻辑回归的功能重要性

时间:2019-12-23 13:52:50

标签: python pyspark logistic-regression

我在PySpark中使用逻辑回归。分割训练和测试数据集后我有

esnext.bigint

我显示了lib,但得到了一个巨大的矩阵。是否有例程选择重要功能并获取其相关列的名称?

1 个答案:

答案 0 :(得分:-1)

可以通过模型coefficients的相对值来推断特征重要性。

>>> model.coefficients
DenseVector([0.0929, -0.3322, -0.7111])
相关问题