Saltar al contenido

¿Cómo guardar model.summary() en un archivo en Keras?

Si te encuentras con algo que te causa duda puedes dejarlo en la sección de comentarios y te ayudaremos lo mas rápido que podamos.

Solución:

Si desea el formato de resumen, puede pasar un print función a model.summary() y la salida al archivo de esa manera:

def myprint(s):
    with open('modelsummary.txt','w+') as f:
        print(s, file=f)

model.summary(print_fn=myprint)

Alternativamente, puede serializarlo en un json o yaml string con model.to_json() o model.to_yaml() que se puede volver a importar más tarde.

Editar

Una forma más pitónica de hacer esto en Python 3.4+ es usar contextlib.redirect_stdout

from contextlib import redirect_stdout

with open('modelsummary.txt', 'w') as f:
    with redirect_stdout(f):
        model.summary()

Aquí tienes otra opción:

with open('modelsummary.txt', 'w') as f:

    model.summary(print_fn=lambda x: f.write(x + 'n'))

Reseñas y puntuaciones

Si conservas alguna duda o disposición de mejorar nuestro reseña te evocamos realizar una nota y con deseo lo ojearemos.

¡Haz clic para puntuar esta entrada!
(Votos: 0 Promedio: 0)



Utiliza Nuestro Buscador

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *