Solucionado (ver solução)
Solucionado
(ver solução)
3
respostas

Como salvar um modelo de rede neural já treinado? (com python)

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!

3 respostas
solução!

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?