Spaces:
Runtime error
Runtime error
santanupoddar
commited on
Commit
•
38a82a5
1
Parent(s):
64549d4
updated epochs to 10
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def malware_detection_DL (results, malicious_traffic, benign_traffic):
|
|
51 |
metrics = ['accuracy'])
|
52 |
#model.fit(x_train, y_train, batch_size=32, epochs = 150, validation_data=(x_test, y_test))
|
53 |
#does not output epochs and gives evalutaion of validation data and history of losses and accuracy
|
54 |
-
history = model.fit(x_train, y_train, batch_size=32, epochs =
|
55 |
_, accuracy = model.evaluate(x_train, y_train)
|
56 |
#return history.history
|
57 |
if results=="Accuracy":
|
|
|
51 |
metrics = ['accuracy'])
|
52 |
#model.fit(x_train, y_train, batch_size=32, epochs = 150, validation_data=(x_test, y_test))
|
53 |
#does not output epochs and gives evalutaion of validation data and history of losses and accuracy
|
54 |
+
history = model.fit(x_train, y_train, batch_size=32, epochs = 10,verbose=0, validation_data=(x_test, y_test))
|
55 |
_, accuracy = model.evaluate(x_train, y_train)
|
56 |
#return history.history
|
57 |
if results=="Accuracy":
|