/Python/Export keras model at specific epoch

Export keras model at specific epoch

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))

Source: stackoverflow.com

Export keras model at specific epoch

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)

-1

Source: stackoverflow.com

Similar codes

3 votes210 views2 answers
Python try without except
Python
31 votes192 views6 answers
Round off python
Python
0 vote181 views1 answer
How to read linux environment variable in python
Python
5 votes271 views2 answers
Selenium.common.exceptions.SessionNotCreatedException: Message: session not created: This version of ChromeDriver only supports Chrome version 81
Python
0 vote214 views1 answer
Matplotlib giving error "OverflowError: In draw_path: Exceeded cell block limit"
Python
-1 vote209 views1 answer
ValueError: cannot reshape array of size 98292 into shape (16382,1,28) site:stackoverflow.com
Python
0 vote147 views1 answer
Django templates directory
Python
0 vote237 views1 answer
Bootstrap 5 in Django
Python Django
0 vote207 views1 answer
Github blxo
Python
1 vote190 views1 answer
How to check any script is running in background linux using python
Python