Demo entry 6796416

Code description teste

   

Submitted by Samuel on May 14, 2019 at 23:09
Language: Python 3. Code size: 937 Bytes.

from sklearn import datasets
from sklearn.metrics import confusion_matrix
from sklearn.model_selection import train_test_split

wine = datasets.load_wine() # Carregando a base de dados e salvando na variável wine

dados = wine.data
rotulos = wine.target # Saída
print(wine.feature_names)
print(wine.target_names)

# Dividindo a base e os rótulos em treino e teste
dados_treino, dados_teste, rotulos_treino, rotulos_teste = train_test_split(dados, rotulos, random_state = 0)

# Treinamento (Árvore de Decisão)
from sklearn.tree import DecisionTreeClassifier
modelo = DecisionTreeClassifier(max_depth = 2).fit(dados_treino, rotulos_treino)

# Usando/Testando o modelo (Predição)
predicao = modelo.predict(dados_teste)
precisao = modelo.score(dados_teste, rotulos_teste)
print('Precisão: ', precisao)

# Criando uma métrica (matriz de confusão)
confusao = confusion_matrix(rotulos_teste, predicao)
print(confusao)

This snippet took 0.01 seconds to highlight.

Back to the Entry List or Home.

Delete this entry (admin only).