[PYTHON] Visualisez l'arbre de décision avec le notebook Jupyter

Visualisez l'arbre de décision avec le notebook Jupyter

import numpy as np
import pydotplus
from sklearn.feature_extraction.text import TfidfVectorizer
from IPython import display
from sklearn import tree
from io import StringIO
from operator import itemgetter
%matplotlib inline
%load_ext autoreload
%autoreload 2



input_text=np.array([
    'Physique Mathématiques Japonais', #Document 1
    'Pan de riz au chocolat', #Document 2
    'Immeuble appartement', #Document 3
    'Hokkaido Fukuoka Kanagawa' #Document 4
])

#Le document 1 vaut 1
#Le document 2 vaut 2
#Le document 3 vaut 3
#Le document 4 vaut 4

target_data = np.array([
    1,2,3,4
    ])

#Document tf-Convertir en vecteur idf
vectorizer=TfidfVectorizer()

input_data=vectorizer.fit_transform(input_text)

np.set_printoptions(precision=2) #Remplacez le nombre de chiffres d'affichage valides par 2 chiffres
print(input_data.toarray())

#Générer un modèle d'arbre de décision
model = tree.DecisionTreeClassifier(max_depth=5)
model = model.fit(input_data, target_data)

model.predict(input_data)

feature_names = list(map(itemgetter(0),
sorted(vectorizer.vocabulary_.items(), key=itemgetter(1))))

data = StringIO()
tree.export_graphviz(model, out_file=data, feature_names=feature_names)
graph = pydotplus.graph_from_dot_data(data.getvalue())

display.display(display.Image(graph.create_png()))

Ça ressemble à ça. Ça fait du bien de voir le compte de Gini!

スクリーンショット 2017-06-02 14.19.14.png

Recommended Posts

Visualisez l'arbre de décision avec le notebook Jupyter
Visualisez l'arbre de décision scikit-learn avec Treemap de Plotly
Utilisation de Graphviz avec Jupyter Notebook
Utiliser pip avec Jupyter Notebook
Utiliser Cython avec Jupyter Notebook
Jouer avec Jupyter Notebook (IPython Notebook)
Autoriser les connexions externes avec le notebook Jupyter
Formatage avec autopep8 sur le notebook Jupyter
Visualisez les résultats des arbres de décision réalisés avec Python scikit-learn
Faites un son avec le notebook Jupyter
Utiliser Markdown avec le notebook Jupyter (avec raccourci)
Ajouter plus de noyaux avec Jupyter Notebook
Analyse pratique avec Pandas + notebook Jupyter
Utiliser nbextensions avec le notebook Jupyter d'Anaconda
Utilisation d'Apache Spark avec le notebook Jupyter (notebook IPython)
Je veux écrire un blog avec Jupyter Notebook
Utiliser Jupyter Lab et Jupyter Notebook avec EC2
Essayez SVM avec scikit-learn sur Jupyter Notebook
Comment utiliser le notebook Jupyter avec ABCI
Lier Python et JavaScript avec le notebook Jupyter
[Mémo Jupyter Notebook] Afficher les kanji avec matplotlib
Sortie de cellule riche avec Jupyter Notebook (IPython)
Mémo Jupyter Notebook
Présentation de Jupyter Notebook
Puissant ordinateur portable Jupyter
Mot de passe du notebook Jupyter
Mémo Jupyter Notebook
J'ai essayé de visualiser tous les arbres de décision de la forêt aléatoire avec SVG
Comment déboguer avec Jupyter ou iPython Notebook
Construire un environnement d'analyse avec Docker (jupyter notebook + PostgreSQL)
Analyse des données pour améliorer POG 2 ~ Analyse avec le notebook jupyter ~
Vérifiez la précision du NLC avec le bloc-notes Jupyter de Watson Studio
Activer Jupyter Notebook avec conda sur un serveur distant
Essayez d'utiliser l'environnement virtuel conda avec Jupyter Notebook
Visualisez de manière interactive les données avec Treasure Data, Pandas et Jupyter.
Remplissez la largeur du bloc-notes Jupyter pour remplir le navigateur
Dessin graphique avec jupyter (notebook ipython) + matplotlib + vagrant
Sélection multiple avec Jupyter
Bougies avec plotly + Jupyter
Machine learning facile avec AutoAI (partie 4) Jupyter Notebook
Visualisez rapidement avec les pandas
Démarrer Jupyter Notebook
Surveiller le modèle d'entraînement avec TensorBord sur Jupyter Notebook
Visualisez les réclamations avec l'IA
Visualisez le nem 2019 avec WordCloud
Dessiner une structure arborescente avec D3.js dans Jupyter Notebook
Importer des cellules spécifiques à partir d'autres blocs-notes avec le bloc-notes Jupyter
Approvisionnement EC2 avec Vagrant + Jupyter (IPython Notebook) sur Docker
3 astuces du notebook Jupyter (Python)
Utilisez nim avec Jupyter
[Cloud103] # 3 Jupyter Notebook à nouveau
Apprentissage automatique à l'aide de Jupyter Notebook dans l'environnement OCI Always Free (2019/12/17)
Refléter l'environnement virtuel créé par Miniconda dans le notebook Jupyter
Jour 65 (Solution) Le bloc-notes Jupyter ne fonctionne pas avec le noyau non connecté.
Je souhaite utiliser un environnement virtuel avec jupyter notebook!
La façon habituelle d'ajouter un noyau avec Jupyter Notebook
Connectez le noyau Jupyter Notebook à Spyder avec Jupytext activé
Écrivez des graphiques en temps réel avec Matplotlib sur le notebook Jupyter
Essayez le clustering avec un modèle gaussien mixte sur Jupyter Notebook