Spaces:
Runtime error
Runtime error
Update main.py
Browse files
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',
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
634 |
model.summary()
|
635 |
|
636 |
#67248
|
@@ -668,7 +679,7 @@ model.summary()
|
|
668 |
|
669 |
# %%
|
670 |
|
671 |
-
log_dir = f"logs/
|
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 |
|