Графвиз от sklearn не может найти мой файл? - PullRequest
0 голосов
/ 16 апреля 2020

Итак, я пытаюсь построить дерево решений в python, используя sklearn.

from sklearn.tree import DecisionTreeClassifier, export_graphviz
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
import graphviz

cancer = load_breast_cancer()
X_train, X_test, y_train, y_test = train_test_split(cancer.data, cancer.target,
                                                    stratify=cancer.target, random_state=42)
tree = DecisionTreeClassifier(random_state=0, max_depth=4)
tree.fit(X_train, y_train)

export_graphviz(tree, out_file=r"C:\Users\obaro\OneDrive\Documents\tree.dot", class_names=["malignant", "benign"],
               feature_names=cancer.feature_names, impurity=False, filled=True)

with open(r"C:\Users\obaro\OneDrive\Documents\tree.dot") as f:
    dot_graph = f.read()
    display(graphviz.Source(dot_graph))

Когда я пытаюсь запустить этот код в Jupyter, я получаю ошибку FileNotFound и ошибку ExecutableNotFound. Сначала я попытался использовать относительный путь, но это не сработало, поэтому я попытался использовать абсолютный путь. Файл был создан и находится в моем текущем домашнем каталоге, поэтому я не уверен, что здесь происходит. Любая помощь будет оценена, спасибо.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...