Regressão de Árvore de Decisão - Intuição e prática
Categories: Machine Learning
Olá meu povo!
Aqui estão o código e dataset utilizados no video:
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Mar 9 18:32:09 2021
@author: rafaeldontalgoncalez
"""
######################################
# Importando as bibliotecas
######################################
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.tree import DecisionTreeRegressor
######################################
# Importa o dataset
######################################
dataset = pd.read_csv('efetividade_vacina.csv')
X = dataset.iloc.values.reshape(-1,1)
y = dataset.iloc.values.reshape(-1,1)
######################################
# Treinando o modelo para regressao de arvore de decisao
######################################
regressor = DecisionTreeRegressor(min_samples_split=10)
regressor.fit(X,y)
######################################
# Imprime a regressao de arvore de decisao
######################################
plt.scatter(X, y, color = 'red')
plt.plot(X, regressor.predict(X), color = 'blue')
plt.title('Regressao de Arvore de Decisao')
plt.xlabel('Idade')
plt.ylabel('Efetividade da Vacina (%)')
plt.show()
######################################
# Prevendo resultados para regressao de arvore de decisao
######################################
print('5 anos =', regressor.predict(]))
print('6 anos =', regressor.predict(]))
print('8 anos =', regressor.predict(]))
print('11 anos =', regressor.predict(]))
print('12 anos =', regressor.predict(]))
print('20 anos =', regressor.predict(]))
print('27 anos =', regressor.predict(]))
print('28 anos =', regressor.predict(]))
print('31 anos =', regressor.predict(]))
print('33 anos =', regressor.predict(]))
print('37 anos =', regressor.predict(]))
print('43 anos =', regressor.predict(]))
print('47 anos =', regressor.predict(]))
print('48 anos =', regressor.predict(]))
print('78 anos =', regressor.predict(]))
