/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

0 vote122 views1 answer
Python turn list of strings into list of doubles
Python
18 votes121 views4 answers
Edit path variable using python
Python
75 votes126 views2 answers
Python find item in list
Python
0 vote166 views1 answer
Python mongodb docker
Python
1 vote151 views1 answer
How to download excel file from s3 using python
Python
0 vote142 views3 answers
How to use css in php example
Python
2 votes164 views2 answers
How to get only first record in django
Python Django
12 votes141 views2 answers
Online ide for python 3
Python
0 vote125 views1 answer
Split coumn of df into multiple dynamic columns
Python
0 vote144 views1 answer
TypeError at /admin/auth/user/
Python