Definir parâmetros e ajustar um modelo
Tarefas preditivas se encaixam em duas categorias: regressão ou classificação. No conjunto de dados de doces, o desfecho é uma variável contínua que descreve com que frequência um doce foi escolhido em vez de outro em uma série de disputas 1 contra 1. Para prever esse valor (a porcentagem de vitórias), você vai usar um modelo de regressão.
Neste exercício, você vai especificar alguns parâmetros usando um modelo de random forest para regressão, rfr.
Este exercício faz parte do curso
Validação de Modelos em Python
Instruções do exercício
- Adicione um parâmetro a
rfrpara que o número de árvores construídas seja 100 e a profundidade máxima dessas árvores seja 6. - Garanta a reprodutibilidade do modelo definindo o estado aleatório como
1111. - Use o método
.fit()para treinar o modelo de random forest de regressão comX_traincomo dados de entrada ey_traincomo resposta.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)