“Ekspor model keras pada zaman tertentu” Kode Jawaban

Ekspor model keras pada zaman tertentu

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

Ekspor model keras pada zaman tertentu

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

Jawaban yang mirip dengan “Ekspor model keras pada zaman tertentu”

Pertanyaan yang mirip dengan “Ekspor model keras pada zaman tertentu”

Lebih banyak jawaban terkait untuk “Ekspor model keras pada zaman tertentu” di Python

Jelajahi jawaban kode populer menurut bahasa

Jelajahi bahasa kode lainnya