dtreeviz is not defined について

Question

前提

dtreevizを用いた決定木分析を行っている。

実現したいこと

ここに実現したいことを箇条書きで書いてください。

• dtreeviz()を使えるようにしたい

発生している問題・エラーメッセージ

NameError: name 'dtreeviz' is not defined

該当のソースコード

from sklearn import tree
from dtreeviz.trees import *
import graphviz

df_past = pd.read_csv(r"C:\Users\Yasu\Documents\情報科学特別演習\csv\スパム検出済み.csv")

X_name = ["reputation","age","sex"]#説明変数
y_name = "spam"#目的変数
X = df_past[X_name]
y = df_past[y_name]

dtree = tree.DecisionTreeClassifier(max_depth=2)
dtree.fit(X,y)

viz = dtreeviz(dtree,X,y,
               target_name = y_name,
               feauture_name = X_name,
               class_names = ["not spam","spam"])

print(viz)

試したこと

・pipを更新して再インストールした。
・graphvizのPATHを確認した。

いずれも改善されませんでした。

補足情報（FW/ツールのバージョンなど）

windows11
IDLE

Comments

[melian]

dtreeviz 関数は dtreeviz.compatibility モジュールに含まれているので、以下の様にして import して試してみて下さい。

from dtreeviz.compatibility import dtreeviz

[yasutin]

実行してみると
"""
viz = dtreeviz(dtree,X,y,
DeprecationWarning: dtreeviz() function is deprecated starting from version 2.0.
For the same functionality, please use this code instead:
m = dtreeviz.model(...)
m.view()

Warning (from warnings module):
File "C:\Users\Yasu\AppData\Local\Programs\Python\Python310\lib\site-packages\sklearn\base.py", line 450
warnings.warn(
UserWarning: X does not have valid feature names, but DecisionTreeClassifier was fitted with feature names
<dtreeviz.utils.DTreeVizRender object at 0x0000013CDC4F41F0>

"""
とでてきたので、
"""
viz = dtreeviz(dtree,X,y,
target_name = y_name,
feature_names = X_name,
class_names = ["not spam","spam"])

print(viz)
"""

↑この部分を
下のように変えました。

"""
m = dtreeviz.model(dtree,X,y,
target_name = y_name,
feature_names = X_name,
class_names = ["not spam","spam"])
m.view()
"""
しかし、以下のエラーメッセージが表示されました。
"""
m = dtreeviz.model(dtree,X,y,
AttributeError: 'function' object has no attribute 'model'
"""
何がいけないのでしょうか。。。

[yasutin]

dtreeviz.model()の引数が違うのでしょうか。

[melian]

どうやら dtreeviz のバージョンアップで API が変更になった様ですね。先程の、

from dtreeviz.compatibility import dtreeviz

を削除して、単に

import dtreeviz

として、m = dtreeviz.model(dtree,X,y, ...) としてみて下さい。

[yasutin]

"""
UserWarning: X does not have valid feature names, but DecisionTreeClassifier was fitted with feature names
"""
このようなエラーメッセージが出てきましたが、X = df_past[X_name]
としているのがいけないのでしょうか？

[melian]

調べてみると、X がデータフレームの場合に、そのワーニングメッセージが表示される様です。なので、X を X.values に変更して試してみて下さい。

[yasutin]

おそらくできたようです！
青文字で下のように出てきたのですが、これはどのようにすれば閲覧できますか？

<dtreeviz.trees.DTreeVizAPI object at 0x000002536E82B550>

[melian]

view メソッドがあるので、

viz.view()

としてみて下さい。

[yasutin]

viewメソッドを実行したものをshowメソッドで実行するらしいです。
なので
"""
m_view = m.view()
m_view.show()
"""
となります。これでできました！

[yasutin]

ベストアンサーにしたいので回答よろしくお願いします。

[melian]

ありがとうございます、回答しました。

Answer 1

※ コメントから転記

dtreeviz のバージョンアップに伴って API が変更になった様なので、以下の様にします。

import dtreeviz

m = dtreeviz.model(dtree, X.values, y, ...)

X がデータフレームの場合に、以下のワーニングメッセージが表示されるので X.values(numpy.ndarray)を渡します。

UserWarning: X does not have valid feature names, but DecisionTreeClassifier was fitted with feature names