pythonstudy Day36

官方文档的阅读

@疏锦行

import pandas as pd import numpy as np from sklearn.datasets import load_iris from sklearn.ensemble import RandomForestClassifier from pdpbox import pdp import matplotlib.pyplot as plt import plotly.io as pio pio.renderers.default="browser"

iris=load_iris()X=pd.DataFrame(iris.data,columns=iris.feature_names)y=iris.target features=iris.feature_names

model=RandomForestClassifier(n_estimators=100,random_state=42)model.fit(X,y)

from pdpbox import pdp pdp_iso=pdp.PDPIsolate(model=model,df=X,model_features=features,feature='petal length (cm)',# 必须是列名字符串 feature_name='petal length (cm)'# 和 feature 保持一致)

fig,_=pdp_iso.plot()fig.show()