我想为我的数据可视化显示决策树图。但是控制台中出现了一个错误
AttributeError: module 'sklearn.tree' has no attribute 'plot_tree'
尽管我通过!pip install -U scikit-learn
和!pip install --upgrade sklearn
安装了额外的模块,但这个错误无法解决
我如何解决这个问题
下面是我的代码
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn.model_selection import train_test_split
# from sklearn.tree import DecisionTreeClassifier
from sklearn import tree
data = pd.read_csv("files/data.csv")
data.drop(["id","Unnamed: 32"],axis=1,inplace=True)
data.diagnosis = [1 if each == "M" else 0 for each in data.diagnosis]
y = data["diagnosis"].values
x_data = data.drop(["diagnosis"],axis=1)
x = (x_data - np.min(x_data))/(np.max(x_data)-np.min(x_data))
x_train, x_test, y_train, y_test = train_test_split(x,y,test_size = 0.3,random_state=1)
dt = DecisionTreeClassifier()
dt.fit(x_train,y_train)
print("score: ", dt.score(x_test,y_test))
tree.plot_tree(dt,
feature_names = data.columns,
rounded = True,
filled = True,
class_names = ["diagnosis"],
impurity = True)
这是我的答案
在更新了anacondanavigator中定义的所有包之后,我编写的代码片段工作得非常完美
相关问题 更多 >
编程相关推荐