Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from __future__ import absolute_import, division, print_function, unicode_literals
- # TensorFlow and tf.keras
- import tensorflow as tf
- from tensorflow import keras
- # Helper libraries
- import numpy as np
- import matplotlib.pyplot as plt
- print(tf.__version__)
- import tensorflowjs
- fashion_mnist = keras.datasets.fashion_mnist
- (train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()
- model = keras.Sequential([
- keras.layers.Flatten(input_shape=(28, 28)),
- keras.layers.Dense(128, activation=tf.nn.relu),
- keras.layers.Dense(10, activation=tf.nn.softmax)
- ])
- model.compile(optimizer='adam',
- loss='sparse_categorical_crossentropy',
- metrics=['accuracy'])
- train_images = train_images / 255.0
- test_images = test_images / 255.0
- model.summary()
- model.fit(train_images, train_labels, epochs=5)
- tensorflowjs.converters.save_keras_model(model, './my_keras_model')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement