import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline
def plot_feature_importance(model):
importance = pd.DataFrame({'Feature': list(X), 'Feature importance': model.feature_importances_})
rank = importance.sort_values(by='Feature importance',ascending=False).head(30)
plt.figure(figsize = (10,12))
plt.title('Top 30 - Features importance - XGBoost',fontsize=14)
s = sns.barplot(y='Feature',x='Feature importance',data=importance, orient='h')
s.set_xticklabels(s.get_xticklabels(),rotation=90)
plt.show()
plot_feature_importance(model)
Привет, я пытаюсь получить важность для набора данных (истощение IBM HR от kaggle), и я получаю ошибку, когда пытаюсь создать график. Вот сообщение об ошибке:
ValueError: все массивы должны быть одинаковой длины