0

Kaggle Titanic データセットに取り組んでいます。私は決定木をよりよく理解しようとしています。私は線形回帰をかなり扱ってきましたが、決定木は決して扱いませんでした。ツリーの python でビジュアライゼーションを作成しようとしています。しかし、何かが機能していません。以下のコードを確認してください。

import pandas as pd
from sklearn import tree
from sklearn.datasets import load_iris
import numpy as np


train_file='.......\RUN.csv'
train=pd.read_csv(train_file)

#impute number values and missing values
train["Sex"][train["Sex"] == "male"] = 0
train["Sex"][train["Sex"] == "female"] = 1
train["Embarked"] = train["Embarked"].fillna("S")
train["Embarked"][train["Embarked"] == "S"]= 0
train["Embarked"][train["Embarked"] == "C"]= 1
train["Embarked"][train["Embarked"] == "Q"]= 2
train["Age"] = train["Age"].fillna(train["Age"].median())
train["Pclass"] = train["Pclass"].fillna(train["Pclass"].median())
train["Fare"] = train["Fare"].fillna(train["Fare"].median())

target = train["Survived"].values
features_one = train[["Pclass", "Sex", "Age", "Fare","SibSp","Parch","Embarked"]].values


# Fit your first decision tree: my_tree_one
my_tree_one = tree.DecisionTreeClassifier(max_depth = 10, min_samples_split = 5, random_state = 1)

iris=load_iris()

my_tree_one = my_tree_one.fit(features_one, target)

tree.export_graphviz(my_tree_one, out_file='tree.dot')

決定木を実際に見るにはどうすればよいですか? それを視覚化しようとしています。

助けてください！

4

3 に答える 3

2

確認しましたか: http://scikit-learn.org/stable/modules/tree.htmlには、ツリーを PNG イメージとしてプロットする方法が記載されています。

 from IPython.display import Image 
 import pydotplus
 dot_data = tree.export_graphviz(my_tree_one, out_file='tree.dot')  
 graph = pydotplus.graph_from_dot_data(dot_data)  `
 Image(graph.create_png())
于 2016-12-15T14:29:12.207 に答える
0

ウィキペディアから：

DOT 言語はグラフを定義しますが、グラフをレンダリングする機能は提供しません。DOT 言語でグラフをレンダリング、表示、および操作するために使用できるプログラムがいくつかあります。

Graphviz - グラフを操作およびレンダリングするためのライブラリとユーティリティのコレクション.

Canviz - ドット ファイルをレンダリングするための JavaScript ライブラリ。

Viz.js - シンプルな Graphviz JavaScript クライアント

Grappa - Graphviz の Java への部分的な移植.[4][5]

Beluging - DOT および Beluga 拡張機能の Python および Google Cloud ベースのビューアー. [1]

Tulip は分析用にドット ファイルをインポートできます

OmniGraffle は DOT のサブセットをインポートして、編集可能なドキュメントを作成できます。(ただし、結果を DOT にエクスポートすることはできません。)

ZGRViewer、GraphViz/DOT Viewer リンク

VizierFX、Flex グラフ レンダリング ライブラリへのリンク

Gephi - あらゆる種類のネットワークと複雑なシステム、動的および階層グラフのためのインタラクティブな視覚化および探索プラットフォーム

したがって、これらのプログラムのいずれかで、ツリーを視覚化できます。

于 2016-12-15T14:36:40.203 に答える