Skip to content

Commit

Permalink
training model complete
Browse files Browse the repository at this point in the history
  • Loading branch information
namannarula committed May 20, 2021
1 parent 5036586 commit fe32cdb
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions RockPaperScissors.py
Original file line number Diff line number Diff line change
Expand Up @@ -241,13 +241,6 @@ def augment_data(image, label):
metrics=['accuracy']
)

# ----- Training the model -----
steps_per_epoch = NUM_TRAIN_EXAMPLES // BATCH_SIZE
validation_steps = NUM_TEST_EXAMPLES // BATCH_SIZE

print('steps_per_epoch:', steps_per_epoch)
print('validation_steps:', validation_steps)

#Defining Callbacks

DESIRED_ACCURACY = 0.90
Expand All @@ -260,6 +253,14 @@ def on_epoch_end(self, epoch, logs={}):

callbacks = myCallback()


# ----- Training the model -----
steps_per_epoch = NUM_TRAIN_EXAMPLES // BATCH_SIZE
validation_steps = NUM_TEST_EXAMPLES // BATCH_SIZE

print('steps_per_epoch:', steps_per_epoch)
print('validation_steps:', validation_steps)

training_history = model.fit(
x=dataset_train_augmented_shuffled.repeat(),
validation_data=dataset_test_shuffled.repeat(),
Expand All @@ -268,5 +269,4 @@ def on_epoch_end(self, epoch, logs={}):
validation_steps=validation_steps,
verbose=1,
callbacks=[callbacks]
)

)

0 comments on commit fe32cdb

Please sign in to comment.