Skip to content

Instantly share code, notes, and snippets.

@HSShashank

HSShashank/train.py Secret

Created August 25, 2021 05:41
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save HSShashank/9dba7a34eb6cf16e70105e66d020a1c3 to your computer and use it in GitHub Desktop.
Save HSShashank/9dba7a34eb6cf16e70105e66d020a1c3 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train.py
#Hyperparameters
LOSS = 'categorical_crossentropy'
OPTIMIZER = 'adam'
BATCH_SIZE = 64
EPOCHS = 20
#Training
model.compile(loss=LOSS, optimizer=OPTIMIZER, metrics=['accuracy'])
history = model.fit(x=X, y=y, batch_size=BATCH_SIZE, epochs=EPOCHS, validation_split=0.2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment