1
resposta

Plot da reta de regressão do modelo treinado

Na última etapa, em que vamos fazer os exercícios e treinar um novo modelo, vamos até salvar o modelo, fica pensando se não seria interessante fazer um plot com a reta de regressão dos dados gerados pelo modelo nos dados reais, e não apenas o plot ("Preço dos Imóveis - Real X Preço dos Imóveis - Previsto") por exemplo. Como poderia fazer este plot da reta de regressão de acordo com o modelo que foi treinado?

1 resposta

Olá, Rafael, tudo bem?

Desculpe pela demora em obter retorno.

Explorar um gráfico com a representação da reta de regressão, que compara os dados gerados pelo modelo aos valores reais, é uma maneira eficaz de avaliar o desempenho do seu modelo.

Para fazer o plot da reta de regressão dos dados gerados pelo modelo nos dados reais, você pode utilizar a biblioteca matplotlib. Importe o matplotlib no seu código utilizando import matplotlib.pyplot as plt.

Em seguida, colete os dados reais que você deseja comparar com as previsões do modelo. Utilize o modelo treinado para fazer as previsões com base nos dados reais. Para isso você pode utilizar a função plot() do matplotlib para criar o gráfico da reta de regressão. Passe os dados reais e as previsões do modelo como argumentos para essa função.

Para calcular os coeficientes da reta de regressão, você pode utilizar a função polyfit() do NumPy.

Abaixo deixo os links da documentação:

Espero ter ajudado.

Caso surja alguma dúvida, é só postar no fórum.

Abraços e até mais!

Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!

Quer mergulhar em tecnologia e aprendizagem?

Receba a newsletter que o nosso CEO escreve pessoalmente, com insights do mercado de trabalho, ciência e desenvolvimento de software