File size: 391 Bytes
d88dd3f a16a0bf d88dd3f 68e8285 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import tensorflow as tf
from model import create_model
import numpy as np
def load_data():
return np.random.rand(1000,64,64,3).astype(np.float32), np.random.randint(10, size=1000)
def main():
train_images, train_labels=load_data()
model = create_model()
model.fit(train_images, train_labels, epochs=5)
model.save('my_model.keras')
if __name__ == "__main__":
main() |