enemy7 commited on
Commit
046e8b1
·
1 Parent(s): 230bd3f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -683,8 +683,8 @@ class MyCustomCallback(tf.keras.callbacks.Callback):
683
 
684
  k = model.fit(X_train,
685
  y_train,
686
- epochs=8,
687
- batch_size=4000,
688
  validation_data=(X_test, y_test),
689
  callbacks=[MyCustomCallback(),tf.keras.callbacks.EarlyStopping(monitor='val_loss', mode='min', verbose=1, patience=10),tensorboard_callback],
690
  verbose=1
@@ -806,8 +806,8 @@ model.save("spam-model.h5",include_optimizer=True)
806
  model.evaluate([X_test],y_test)
807
 
808
  import time
809
- print("MODELLLLLLL save ")
810
- os.system('cp log.txt old.txt')
811
 
812
  while 1:
813
  print("waiting")
 
683
 
684
  k = model.fit(X_train,
685
  y_train,
686
+ epochs=10,
687
+ batch_size=512,
688
  validation_data=(X_test, y_test),
689
  callbacks=[MyCustomCallback(),tf.keras.callbacks.EarlyStopping(monitor='val_loss', mode='min', verbose=1, patience=10),tensorboard_callback],
690
  verbose=1
 
806
  model.evaluate([X_test],y_test)
807
 
808
  import time
809
+ print("\n\nMODELLLLLLL save ")
810
+ # os.system('cp log.txt old.txt')
811
 
812
  while 1:
813
  print("waiting")