Advertisement
Guest User

Untitled

a guest
May 24th, 2019
68
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.98 KB | None | 0 0
  1. import tensorflow as tf
  2.  
  3. class ACCCallback(tf.keras.callbacks.Callback):
  4. def on_epoch_end(self, epoch, logs={}):
  5. if logs.get('acc')>0.8:
  6. print('\n Reach 80% accuracy, so early stop.')
  7. self.model.stop_training = True
  8.  
  9.  
  10. (x_train, y_train), (x_test, y_test) = tf.keras.datasets.fashion_mnist.load_data()
  11. x_train,x_test = x_train/255., x_test/255.
  12. acc_callback = ACCCallback()
  13. model = keras.Sequential([keras.layers.Flatten(),
  14. keras.layers.Dense(128,activation=tf.nn.relu),
  15. keras.layers.Dense(10,activation='softmax')])
  16. model.compile(loss='sparse_categorical_crossentropy',optimizer=tf.train.AdamOptimizer(),metrics=['accuracy'])
  17. model.fit(x_train,y_train,epochs=10,callbacks=[acc_callback])
  18.  
  19. """
  20. Epoch 1/10
  21. 59744/60000 [============================>.] - ETA: 0s - loss: 0.4633 - acc: 0.8371Reach 80% accuracy, so early stop.
  22. 60000/60000 [==============================] - 6s 106us/sample - loss: 0.4630 - acc: 0.8372
  23. """
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement