/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 votes106 views1 answer
Change graph colors python matplotlib
Python
0 vote109 views1 answer
QuizListView login required django
Python
5 votes103 views2 answers
How to empty a text file in python
Python
0 vote103 views1 answer
Create jwt token python
Python
23 votes104 views1 answer
Python minute from datetime
Python
94 votes102 views5 answers
How to rename columns using panda object
Python
12 votes104 views4 answers
Python join dict
Python
1 vote104 views1 answer
Django sessions for beginners
Python
17 votes103 views2 answers
Count lines in file python
Python
-1 vote100 views1 answer
Python squared
Python