安装dtreeviz后调用出现NameError错误求助

问题根源 #

你用的from dtreeviz.trees import *只导入了子模块里的内容，但dtreeviz这个核心绘图函数并不在该子模块中，需要从主模块导入才行。

修复代码 #

修改导入语句即可，两种方式任选其一：

方式一：直接导入核心函数 #

from dtreeviz import 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

方式二：导入整个主模块 #

import dtreeviz

viz = dtreeviz.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

额外检查点 #

• 确认Python环境一致：有时Spyder或Jupyter使用虚拟环境，但你用pip将库安装到了全局环境。可在Spyder终端或Jupyter代码块中执行!pip show dtreeviz，对比安装路径与当前环境的Python路径是否匹配。
• 若环境不匹配，直接在Spyder/Jupyter的终端中执行!pip install dtreeviz==2.1.3，将库安装到当前运行环境中。

内容的提问来源于stack exchange，提问作者Tommy