出现 NameError: name 'export_graphviz' is not defined
错误是因为在代码中使用了 export_graphviz
函数,但没有正确导入它。export_graphviz
函数是 sklearn.tree
模块提供的一个函数,用于导出决策树模型的结构为 Graphviz 的 DOT 源码格式。
要解决这个问题,请确保在你的代码文件顶部添加以下导入语句:
from sklearn.tree import export_graphviz
完整的导入语句应该是这样的:
import pandas as pd
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.tree import DecisionTreeClassifier, export_graphviz # 确保这里导入了 export_graphviz
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.preprocessing import LabelEncoder
from sklearn.metrics import accuracy_score
from graphviz import Source
确保你的代码中包含这些导入语句,export_graphviz
函数就可以在你的代码中正常使用了。以下是修正后完整的代码示例:
import pandas as pd
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.tree import DecisionTreeClassifier, export_graphviz # 导入 export_graphviz
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.preprocessing import LabelEncoder
from sklearn.metrics import accuracy_score
from graphviz import Source
# ... 其余的代码 ...
添加正确的导入语句后,再次运行你的代码,NameError
错误应该会消失。