【发布时间】:2021-01-29 18:18:34
【问题描述】:
我一直在使用Py-earth模型,url可以在这里找到:
然后跟随这个 turorial here :
我已经成功地训练了:
model = Earth(max_degree=2, penalty=0.5, minspan_alpha =1, endspan_alpha =1,allow_linear=False,check_every=1,verbose=True)
model.fit(X_train,y_train)
但是当使用model.summary_feature_importances(sort_by='gcv') 时,它会返回一个空数组,它不应该这样做,因为它在教程中提供了这样的结果:
nb_subsets gcv rss
lstat 0.42 0.78 0.78
rm 0.25 0.12 0.12
crim 0.17 0.06 0.06
dis 0.17 0.04 0.04
black 0.00 0.00 0.00
ptratio 0.00 0.00 0.00
tax 0.00 0.00 0.00
rad 0.00 0.00 0.00
age 0.00 0.00 0.00
nox 0.00 0.00 0.00
chas 0.00 0.00 0.00
indus 0.00 0.00 0.00
zn 0.00 0.00 0.00
本教程可以在here找到。
【问题讨论】:
-
在构造
Earth对象时需要使用feature_importance_type参数。见contrib.scikit-learn.org/py-earth/_modules/pyearth/earth.html。
标签: python machine-learning regression