Можно ли печатать дерево решений в scikit-learn?

Есть ли способ напечатать обученное дерево решений в scikit-learn? Я хочу обучить дерево решений для моей диссертации, и я хочу поместить картину дерева в тезис. Это возможно?

One Solution collect form web for “Можно ли печатать дерево решений в scikit-learn?”

Существует способ экспорта в формат graph_viz: http://scikit-learn.org/stable/modules/generated/sklearn.tree.export_graphviz.html

Поэтому из онлайн-документов:

>>> from sklearn.datasets import load_iris >>> from sklearn import tree >>> >>> clf = tree.DecisionTreeClassifier() >>> iris = load_iris() >>> >>> clf = clf.fit(iris.data, iris.target) >>> tree.export_graphviz(clf, ... out_file='tree.dot') 

Затем вы можете загрузить это с помощью графика, или если у вас установлен pydot, вы можете сделать это более непосредственно: http://scikit-learn.org/stable/modules/tree.html

 >>> from sklearn.externals.six import StringIO >>> import pydot >>> dot_data = StringIO() >>> tree.export_graphviz(clf, out_file=dot_data) >>> graph = pydot.graph_from_dot_data(dot_data.getvalue()) >>> graph.write_pdf("iris.pdf") 

Будет генерировать svg, не может отобразить его здесь, поэтому вам нужно будет перейти по ссылке: http://scikit-learn.org/stable/_images/iris.svg

  • sklearn LogisticRegression и изменение порога по умолчанию для классификации
  • Как преобразовать подмножество Pandas DataFrame столбцов И строк в массив numpy?
  • Можно ли использовать scikit из IronPython?
  • Как свернуть собственный оценщик в PySpark mllib
  • Python scikit-learn SVM Classifier "ValueError: найден массив с тусклым 3. Ожидаемый <= 2"
  • Векторная авторегрессионная модель с scikit-learn
  • Python scikit-learn: не может клонировать объект ... поскольку конструктор, похоже, не задает параметр
  • Sklearn - Невозможно использовать закодированные данные в классификаторе случайных лесов
  • scikit изучает создание фиктивных переменных
  • Ошибка в scikit-узнает функцию LDA - графики показывают ненулевую корреляцию
  • Pandas & Scikit: использование памяти при разрезании DataFrame
  • Python - лучший язык программирования в мире.