Dtreeviz 名称错误

Question

我安装了 dtreebiz 并想制作一个三图。我完全按照下面的示例代码进行了操作，但不断收到“NameError：名称‘dtreeviz’未定义”。

from dtreeviz.trees import *

viz = dtreeviz(lgbm,
               x_data = df_X_train,
               y_data = df_y_train,
               target_name = TARGET,
               feature_names = df_X_train.columns.tolist(),
               tree_index = 0)

viz

我使用pip show dtreeviz检查了版本并确认安装了版本 2.1.3。我主要使用 Spyder，但我也用 Jupyter Notebook 尝试过同样的事情，但得到了同样的错误。

Answer 1

你需要添加

import dtreeviz

到您的代码（请参阅 dtreeviz github 页面中的快速入门）。从 dtreeviz.trees 导入所有符号，就像你正在做的那样

from dtreeviz.trees import *

仅导入 trees 模块中的符号。