Usando o sckit-learn e o Keras, como salvar um modelo treinado e "importá-lo" depois?
Assim, não precisaria ficar começando do zero toda vez que rodar o programa!
Usando o sckit-learn e o Keras, como salvar um modelo treinado e "importá-lo" depois?
Assim, não precisaria ficar começando do zero toda vez que rodar o programa!
Olá Edson, tudo bem?
Para isso, você consegue utilizar o Pickle. Com ele, conseguimos guardar um estado de um objeto em um arquivo.
Por exemplo:
# código para criar o modelo omitido
import pickle
modelo.fit(X, y)
meu_arquivo = open('meu_arquivo.p', 'wb')
pickle.dump(modelo, meu_arquivo)
meu_arquivo.close()
Para carregar esse modelo, podemos usar:
import pickle
meu_arquivo = open('meu_arquivo.p', 'rb')
modelo = pickle.load(meu_arquivo)
@Yuri Matheus:
Acho que não deu certo:
Estou recebendo o erro, após carregar o modelo salvo!
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-74-2674d4ee4eaa> in <module>()
----> 1 predictions = logmodel.predict(X_test)
AttributeError: '_io.BufferedReader' object has no attribute 'predict'
Código fonte: https://pastebin.com/kjGyakmC
Opa Edson, tudo bem? Esse tópico já foi fechado, você poderia abrir outro, por favor?