python构建lightgbm模型,制作force plot代码

时间: 2023-06-18 21:05:22 浏览: 72
构建LightGBM模型可以使用以下代码: ```python import lightgbm as lgb from sklearn.datasets import load_boston from sklearn.model_selection import train_test_split # 载入数据 data = load_boston() X_train, X_test, y_train, y_test = train_test_split(data.data, data.target, test_size=0.2, random_state=42) # 构建LightGBM模型 params = { 'boosting_type': 'gbdt', 'objective': 'regression', 'metric': {'l2', 'l1'}, 'num_leaves': 31, 'learning_rate': 0.05, 'feature_fraction': 0.9, 'bagging_fraction': 0.8, 'bagging_freq': 5, 'verbose': 0 } lgb_train = lgb.Dataset(X_train, y_train) lgb_eval = lgb.Dataset(X_test, y_test, reference=lgb_train) gbm = lgb.train(params, lgb_train, num_boost_round=100, valid_sets=lgb_eval, early_stopping_rounds=5) # 预测结果 y_pred = gbm.predict(X_test, num_iteration=gbm.best_iteration) ``` 制作Force Plot需要使用shap包,可以使用以下代码实现: ```python import shap # 计算SHAP值 explainer = shap.TreeExplainer(gbm) shap_values = explainer.shap_values(X_test) # 制作Force Plot shap.initjs() shap.force_plot(explainer.expected_value, shap_values[0,:], X_test[0,:]) ``` 其中,`shap.TreeExplainer()`用于计算SHAP值,`shap_values()`用于计算所有测试数据的SHAP值,`shap.force_plot()`用于制作Force Plot。

相关推荐

最新推荐

recommend-type

python中matplotlib实现随鼠标滑动自动标注代码

Python3.6.5,代码示例: import matplotlib.pyplot as plt import numpy as np def Show(y): #参数为一个list len_y = len(y) x = range(len_y) _y = [y[-1]]*len_y fig = plt.figure(figsize=(960/72,360/72...
recommend-type

Python设置matplotlib.plot的坐标轴刻度间隔以及刻度范围

主要介绍了Python设置matplotlib.plot的坐标轴刻度间隔以及刻度范围,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
recommend-type

Python使用matplotlib绘制圆形代码实例

def plot_circle(center=(3, 3),r=2): x = np.linspace(center[0] - r, center[0] + r, 5000) y1 = np.sqrt(r**2 - (x-center[0])**2) + center[1] y2 = -np.sqrt(r**2 - (x-center[0])**2) + center[1] plt....
recommend-type

python matplotlib绘制动态曲线 数据流可视化

很多时候需要在程序运行过程中,查看一些数据的动态变化,最容易想到的是像opencv那样直接循环使用imshow()形成动态画面,但是由于matplotlib中的显示模式是阻塞模式,在plt.show()之后程序就会暂停在那,打开一个...
recommend-type

python之matplotlib.pyplot迭代累积绘制曲线问题及解决办法

问题来源:python之随机漫步模拟 解决方法:在绘图命令前加pyplot.cla()清除上一个坐标轴或者pyplot.close()直接关闭上一个图表重新制图 更改后: from random import choice as choice import matplotlib.pyplot as...
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

2. 通过python绘制y=e-xsin(2πx)图像

可以使用matplotlib库来绘制这个函数的图像。以下是一段示例代码: ```python import numpy as np import matplotlib.pyplot as plt def func(x): return np.exp(-x) * np.sin(2 * np.pi * x) x = np.linspace(0, 5, 500) y = func(x) plt.plot(x, y) plt.xlabel('x') plt.ylabel('y') plt.title('y = e^{-x} sin(2πx)') plt.show() ``` 运行这段
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。