我正在尝试为xgboost模型获取功能重要性,但我总是收到此错误:
AttributeError: 'XGBModel' object has no attribute 'feature_importances_'
我读到这可能是一个版本问题,但是我在2016年12月下旬安装了这个版本,我相信这是可以修复此问题的版本。
有人可以帮助我获取功能重要性吗?
import numpy as np
import pandas as pd
import xgboost as xgb
print "XGB Version: " + str(xgb.__version__)
np.random.seed(0)
df = pd.DataFrame(np.random.randint(0,1000,size=(1000, 5)), columns=list('ABCDE'))
x = df.ix[:,0:4]
y = df.ix[:,4]
mod = xgb.XGBModel()
mod.fit (x,y)
mod.feature_importances_