/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

1 vote193 views1 answer
Coronavirus program in python
Python
0 vote209 views1 answer
Settings.debug django
Python
0 vote200 views1 answer
Picobot python
Python
0 vote188 views1 answer
How to get session value in django template
Python
0 vote227 views1 answer
Python requests token x-www-form-urlencoded
Python
2 votes208 views3 answers
Miles to km in python
Python
71 votes194 views6 answers
Clear console in python
Python
10 votes226 views3 answers
Isletter python
Python
10 votes212 views3 answers
Dataframein python
Python
27 votes197 views3 answers
Splitting a string and appending each character to a list python
Python