当前位置:  开发笔记 > 编程语言 > 正文

如何使用scikit线性回归找到系数的特征名称?

如何解决《如何使用scikit线性回归找到系数的特征名称?》经验,为你挑选了1个好方法。

诀窍在于,在训练模型后,您就知道系数的顺序:

model_1 = linear_model.LinearRegression()
model_1.fit(train_data[model_1_features], train_data['price'])
print(list(zip(model_1.coef_, model_1_features)))

这将打印系数和正确的功能.(用pandas DataFrame测试)

如果您想稍后重用系数,也可以将它们放在字典中:

coef_dict = {}
for coef, feat in zip(model_1.coef_,model_1_features):
    coef_dict[feat] = coef

(你可以通过训练两个具有相同功能的模型来测试它,但正如你所说的那样,改进了功能的顺序.)



1> Robin Spiess..:

诀窍在于,在训练模型后,您就知道系数的顺序:

model_1 = linear_model.LinearRegression()
model_1.fit(train_data[model_1_features], train_data['price'])
print(list(zip(model_1.coef_, model_1_features)))

这将打印系数和正确的功能.(用pandas DataFrame测试)

如果您想稍后重用系数,也可以将它们放在字典中:

coef_dict = {}
for coef, feat in zip(model_1.coef_,model_1_features):
    coef_dict[feat] = coef

(你可以通过训练两个具有相同功能的模型来测试它,但正如你所说的那样,改进了功能的顺序.)

推荐阅读
135369一生真爱_890
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有