【发布时间】:2021-03-14 16:50:37
【问题描述】:
# use feature importance for feature selection
from numpy import loadtxt
from numpy import sort
from xgboost import XGBClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.feature_selection import SelectFromModel
# load data
dataset = loadtxt('.\\DataSets\\pima-indians-diabetes.csv', delimiter=",")
# split data into X and y
X = dataset[:,0:8]
Y = dataset[:,8]
# split data into train and test sets
X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.33, random_state=7)
# fit model on all training data
model = XGBClassifier(eval_metric = "error")
model.fit(X_train, y_train)
# make predictions for test data and evaluate
y_pred = model.predict(X_test)
predictions = [round(value) for value in y_pred]
accuracy = accuracy_score(y_test, predictions)
print("Accuracy: %.2f%%" % (accuracy * 100.0))
# Fit model using each importance as a threshold
thresholds = sort(model.feature_importances_)
for thresh in thresholds:
# select features using threshold
selection = SelectFromModel(model, threshold=thresh, prefit=True)
select_X_train = selection.transform(X_train)
# train model
selection_model = XGBClassifier(eval_metric = "error")
selection_model.fit(select_X_train, y_train)
# eval model
select_X_test = selection.transform(X_test)
select_X_test = X_test[:, 0:select_X_test.shape[1]]*0 + select_X_test
y_pred = selection_model.predict(select_X_test)
predictions = [round(value) for value in y_pred]
accuracy = accuracy_score(y_test, predictions)
print("Thresh=%.3f, n=%d, Accuracy: %.2f%%" % (thresh, select_X_train.shape[1], accuracy*100.0))
您可以从https://www.kaggle.com/kumargh/pimaindiansdiabetescsv获取数据集
运行上面的代码,结果是:
Accuracy: 74.02%
Thresh=0.088, n=8, Accuracy: 74.02%
Thresh=0.089, n=7, Accuracy: 71.65%
Thresh=0.098, n=6, Accuracy: 71.26%
Thresh=0.098, n=5, Accuracy: 74.41%
Thresh=0.100, n=4, Accuracy: 74.80%
Thresh=0.136, n=3, Accuracy: 71.26%
Thresh=0.152, n=2, Accuracy: 71.26%
Thresh=0.240, n=1, Accuracy: 67.32%
但是,当注释掉 select_X_test = X_test[:, 0:select_X_test.shape[1]]*0 + select_X_test 行时。这条线没有意义,但结果是:
Accuracy: 74.02%
Thresh=0.088, n=8, Accuracy: 60.63%
Thresh=0.089, n=7, Accuracy: 61.02%
Thresh=0.098, n=6, Accuracy: 59.45%
Thresh=0.098, n=5, Accuracy: 57.87%
Thresh=0.100, n=4, Accuracy: 63.39%
Thresh=0.136, n=3, Accuracy: 56.30%
Thresh=0.152, n=2, Accuracy: 57.87%
Thresh=0.240, n=1, Accuracy: 67.32%
有什么区别,一个错误?我认为第一个结果是正确的。
是否设置种子不会影响差异。不同的是 select_X_test = X_test[:, 0:select_X_test.shape[1]]*0 + select_X_test
【问题讨论】:
-
不阅读所有细节,我会从设置种子开始
-
那 lline 不是没有意义的,你用 *0 乘法禁用了一些功能
标签: python machine-learning scikit-learn xgboost