python学习day32

官方文档的阅读
知识点回顾：
1.官方文档的检索方式：github和官网（对应的库）
2.官方文档的阅读和使用：要求安装的包和文档为同一个版本
3.类的关注点：
a.实例化所需要的参数
b.普通方法所需要的参数
c.普通方法的返回值
4.绘图的理解：对底层库的调用

如pdpbox

GitHub 仓库：https://github.com/SauceCat/PDPbox
PyPI 页面：https://pypi.org/project/PDPbox/
官方文档：https://pdpbox.readthedocs.io/en/latest/

import pandas as pd
from sklearn.ensemble import RandomForestClassifier
from pdpbox import pdp_interact, pdp_interact_plot
data = pd.read_csv('titanic.csv')
X = data[['Age', 'Fare', 'Pclass', 'Sex']]
y = data['Survived']
 
# 简单训练一个分类模型
model = RandomForestClassifier()
model.fit(X, y)
interact_features = ['Age', 'Fare']
 
# 实例化pdp_interact类
interact_out = pdp_interact(
    model=model, 
    dataset=X, 
    model_features=X.columns.tolist(), 
    features=interact_features,
    num_grid_points=10  # 每个特征网格点数
)
# 生成热力图（展示二维交互影响）
pdp_plot = pdp_interact_plot(
    pdp_interact_out=interact_out, 
    feature_names=interact_features, 
    plot_type='grid',  # 可选'contour'等高线
    plot_pts_dist=True  # 显示数据点分布
)
 
# 自定义样式（如标题、颜色）
pdp_plot['pdp_interact_plot'].figure.suptitle('Interaction between Age and Fare', y=1.05)
pdp_plot['pdp_interact_plot'].set_colormap('viridis')  # 颜色映射