enemy7 commited on
Commit
4a4ee60
·
1 Parent(s): d9c9e29

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +13 -2
main.py CHANGED
@@ -630,7 +630,18 @@ output_layer = tf.keras.layers.Dense(1,activation='sigmoid',name="Spam")(f1)
630
  # Create the model
631
  model = tf.keras.Model(inputs=[string_input, length_input,currency_input,spam_word_input,emoji_input,contain_input,email_input,phone_input], outputs=output_layer)
632
 
633
- model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
 
 
 
 
 
 
 
 
 
 
 
634
  model.summary()
635
 
636
  #67248
@@ -668,7 +679,7 @@ model.summary()
668
 
669
  # %%
670
 
671
- log_dir = f"logs/fit_{datetime.datetime.now()}/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
672
  tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
673
 
674
 
 
630
  # Create the model
631
  model = tf.keras.Model(inputs=[string_input, length_input,currency_input,spam_word_input,emoji_input,contain_input,email_input,phone_input], outputs=output_layer)
632
 
633
+ model.compile(optimizer='adam', loss='binary_crossentropy',metrics=['accuracy',
634
+ tf.keras.metrics.Precision(),
635
+ tf.keras.metrics.Recall(),
636
+ tf.keras.metrics.TruePositives(),
637
+ tf.keras.metrics.TrueNegatives(),
638
+ tf.keras.metrics.FalsePositives(),
639
+ tf.keras.metrics.FalseNegatives(),
640
+ tf.keras.metrics.AUC(),
641
+ tf.keras.metrics.PrecisionAtRecall(0.5),
642
+ tf.keras.metrics.SensitivityAtSpecificity(0.5),
643
+ tf.keras.metrics.SpecificityAtSensitivity(0.5)]
644
+ )
645
  model.summary()
646
 
647
  #67248
 
679
 
680
  # %%
681
 
682
+ log_dir = f"logs/fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
683
  tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
684
 
685