“Modelo de Keras de exportação em época específica” Respostas de código

Modelo de Keras de exportação em é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 Keras de exportação em é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

Respostas semelhantes a “Modelo de Keras de exportação em época específica”

Perguntas semelhantes a “Modelo de Keras de exportação em época específica”

Mais respostas relacionadas para “Modelo de Keras de exportação em época específica” em Python

Procure respostas de código populares por idioma

Procurar outros idiomas de código