Max Epochs = 10
ADAM
sparse_categorical_crossentropy
import tensorflow as tf
fashion_mnist = tf.keras.datasets.fashion_mnist
class myCallback(tf.keras.callbacks.Callback):
def on_epoch_end(self, epoch, logs={}):
if(logs.get('acc')>0.99):
print("\nReached 99% accuracy so cancelling training!")
self.model.stop_training = True
mnist = tf.keras.datasets.mnist
(x_train, y_train),(x_test, y_test) = fashion_mnist.load_data()
x_train = x_train/255.0
x_test = x_test/255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(100, activation=tf.nn.relu),
tf.keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
callbacks = myCallback()
model.fit(x_train, y_train, epochs=10, callbacks=[callbacks])
ADAM
sparse_categorical_crossentropy
import tensorflow as tf
fashion_mnist = tf.keras.datasets.fashion_mnist
class myCallback(tf.keras.callbacks.Callback):
def on_epoch_end(self, epoch, logs={}):
if(logs.get('acc')>0.99):
print("\nReached 99% accuracy so cancelling training!")
self.model.stop_training = True
mnist = tf.keras.datasets.mnist
(x_train, y_train),(x_test, y_test) = fashion_mnist.load_data()
x_train = x_train/255.0
x_test = x_test/255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(100, activation=tf.nn.relu),
tf.keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
callbacks = myCallback()
model.fit(x_train, y_train, epochs=10, callbacks=[callbacks])
Fashion MNIST
Optimizer: Adam
Loss: Sparse categorical cross entropy
Neural Network
|
Training Set Acc
|
Test Set Accu.
|
Time/Epoch (ADAM, 10 epochs)
|
1024*100*10
|
0.9154
|
0.8855
|
20s
|
512*256*10
|
0.9151
|
0.8867
|
16s (local laptop)
|
1024*10*10*10
|
0.9119
|
0.8869
|
19s
|
1024*10
|
0.9173
|
0.8743
|
24s (local laptop)
|
256*10*10*10*10
|
0.9051
|
0.8745
|
7s
|
2000*10
|
0.9151
|
0.8842
|
31s
|
64*10
|
0.9026
|
0.8821
|
3s (local)
|
10*10
|
0.8641
|
0.8455
|
2s (local)
|
20*10
|
0.8814
|
0.8631
|
3s (local)
|
20*20*10
|
0.8858
|
0.8563
|
3s (local)
|
100*10
|
0.9101
|
0.8789
|
4s
|
Comments
Post a Comment