“Modelo de exportación de Keras en una época específica” Código de respuesta

Modelo de exportación de Keras en una época específica

import keras

class CustomSaver(keras.callbacks.Callback):
    def on_epoch_end(self, epoch, logs={}):
        if epoch == 2:  # or save after some epoch, each k-th epoch etc.
            self.model.save("model_{}.hd5".format(epoch))
ashattack

Modelo de exportación de Keras en una época específica

import keras
import numpy as np

inp = keras.layers.Input(shape=(10,))
dense = keras.layers.Dense(10, activation='relu')(inp)
out = keras.layers.Dense(1, activation='sigmoid')(dense)
model = keras.models.Model(inp, out)
model.compile(optimizer="adam", loss="binary_crossentropy",)

# Just a noise data for fast working example
X = np.random.normal(0, 1, (1000, 10))
y = np.random.randint(0, 2, 1000)

# create and use callback:
saver = CustomSaver()
model.fit(X, y, callbacks=[saver], epochs=5)
ashattack

Respuestas similares a “Modelo de exportación de Keras en una época específica”

Preguntas similares a “Modelo de exportación de Keras en una época específica”

Más respuestas relacionadas con “Modelo de exportación de Keras en una época específica” en Python

Explore las respuestas de código populares por idioma

Explorar otros lenguajes de código