plt.figure(figsize=(15,7.5))
plot_tree(clf_dt, filled= True, `rounded=True, class_names = ["No HD","Yes HD"], feature_names=X_encoded.columns)
我是新来的机器学习和 p。 • 为UCI储存库的心脏病数据绘制分类树。 在我试图种植树木时,在X-encoded对分类变量进行热点编码后,我向我发出这一错误信息:
InvalidParameterError: The feature_names parameter of plot_tree must be an instance of list or None. Got Index([ age , sex , cp , restbp , chol , fbs , restecg , thalach ,
exang , oldpeak , slope , ca , thal ],
dtype= object ) instead.