前提
dtreevizを用いた決定木分析を行っている。
実現したいこと
ここに実現したいことを箇条書きで書いてください。
- dtreeviz()を使えるようにしたい
発生している問題・エラーメッセージ
NameError: name 'dtreeviz' is not defined
該当のソースコード
python
1from sklearn import tree 2from dtreeviz.trees import *3import graphviz 4 5df_past = pd.read_csv(r"C:\Users\Yasu\Documents\情報科学特別演習\csv\スパム検出済み.csv")6 7X_name = ["reputation","age","sex"]#説明変数8y_name = "spam"#目的変数9X = df_past[X_name]10y = df_past[y_name]11 12dtree = tree.DecisionTreeClassifier(max_depth=2)13dtree.fit(X,y)14 15viz = dtreeviz(dtree,X,y,16 target_name = y_name,17 feauture_name = X_name,18 class_names = ["not spam","spam"])19 20print(viz)21
試したこと
・pipを更新して再インストールした。
・graphvizのPATHを確認した。
いずれも改善されませんでした。
補足情報(FW/ツールのバージョンなど)
windows11
IDLE

0 コメント