使用for循环从模型集合绘制变量重要性图

import matplotlib.pyplot as plt
import numpy as np

models = [model1, model2, model3]  # 假设有三个模型

variable_importance = []  # 存储变量重要性

for model in models:
    importance = model.feature_importance()  # 获取模型的变量重要性
    variable_importance.append(importance)

# 计算平均变量重要性
mean_importance = np.mean(variable_importance, axis=0)

# 获取变量名称
variable_names = ['var1', 'var2', 'var3']  # 假设有三个变量

# 绘制柱状图
plt.bar(variable_names, mean_importance)
plt.xlabel('Variable')
plt.ylabel('Importance')
plt.title('Variable Importance')

# 显示图形
plt.show()