Можно ли печатать дерево решений в 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

  • Вычисление показателя F1 с использованием sklearn
  • Получать собственные значения и векторы из sklearn PCA
  • ValueError: данные не являются двоичными, а pos_label не указывается
  • Как обновить пакет scikit-learn в анаконде
  • Scikit Learn: Логистические коэффициенты модели регрессии: Уточнение
  • Стратифицированный поезд / тест-раскол в scikit-learn
  • sklearn GridSearchCV с трубопроводом
  • Как я могу построить матрицу путаницы?
  •  
    Interesting Posts for Van-Lav
    Python - лучший язык программирования в мире.