1
resposta

[Dúvida] Como fazer uma predição com o modelo treinado

Como faço para salvar o modelo treinado para usar em dados e fazer a previsão deles?

1 resposta

Oii, Marcos! Tudo joia?

Como sugestão você pode usar a biblioteca chamada pickle. Ela permite que você salve objetos Python para uso posterior, sem precisar treinar o modelo novamente.

Segue um exemplo para ilustrar como seria:

import pickle

# supondo que 'modelo' é o seu modelo treinado
with open('modelo_treinado.pkl', 'wb') as file:
    pickle.dump(modelo, file)

Com o 'modelo_treinado.pkl' salvo, para carregá-lo novamente, segue um exemplo de código:

with open('modelo_treinado.pkl', 'rb') as file:
    modelo_carregado = pickle.load(file)

Feito isso, pode usar modelo_carregado para fazer previsões, assim como faria com o modelo original. É importante considerar uma adaptação do exemplo acima ao seu projeto e realizar testes para entender como funciona melhor para ele.

E caso queira expandir os seus conhecimentos, recomendo a Formação Machine Learning com Python que tem como pontos principais a classificação, validação e métricas de modelos.

Espero ter ajudado. Qualquer dúvida, compartilhe no fórum para te auxiliarmos.

Bons estudos, Marcos!

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