Como faço para salvar o modelo treinado para usar em dados e fazer a previsão deles?
Como faço para salvar o modelo treinado para usar em dados e fazer a previsão deles?
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!