“Exportieren Sie das Kerasmodell in einer bestimmten Epoche” Code-Antworten

Exportieren Sie das Kerasmodell in einer bestimmten Epoche

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

Exportieren Sie das Kerasmodell in einer bestimmten Epoche

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

Ähnliche Antworten wie “Exportieren Sie das Kerasmodell in einer bestimmten Epoche”

Fragen ähnlich wie “Exportieren Sie das Kerasmodell in einer bestimmten Epoche”

Weitere verwandte Antworten zu “Exportieren Sie das Kerasmodell in einer bestimmten Epoche” auf Python

Durchsuchen Sie beliebte Code-Antworten nach Sprache

Durchsuchen Sie andere Codesprachen