/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 vote381 views1 answer
Flask get uploaded file size
Python
0 vote394 views2 answers
Install wget in anaconda
Python
2 votes367 views1 answer
Add whitespaces between char python
Python
0 vote352 views1 answer
How to stop a program after 1 second in python
Python
1 vote357 views1 answer
How to say hello with name in python
Python
1 vote356 views1 answer
Programe to check if a is divisible
Python
0 vote344 views1 answer
Compile python folder
Python
9 votes301 views1 answer
Python debugger
Python
83 votes339 views6 answers
Py virtual environment
Python
0 vote325 views1 answer
In <lambda> tk.Button(self.frame, text = text, command= lambda:self.new_window(number, _class)).grid(padx= 5, pady= 5, sticky= N + E + W + S) File "C:\Users\CSB\Desktop\python program\triple_window.py", line 50, in new_window
Python