From ecd3c734d00671a4fe5ac1713422dae55f2bad2f Mon Sep 17 00:00:00 2001
From: Luca Deri <deri@ntop.org>
Date: Mon, 28 Oct 2024 12:55:18 +0100
Subject: Rename

---
 dga/tensorflow_tests/test_script.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 dga/tensorflow_tests/test_script.py

(limited to 'dga/tensorflow_tests/test_script.py')

diff --git a/dga/tensorflow_tests/test_script.py b/dga/tensorflow_tests/test_script.py
new file mode 100644
index 000000000..5c946c8cf
--- /dev/null
+++ b/dga/tensorflow_tests/test_script.py
@@ -0,0 +1,22 @@
+import tensorflow as tf
+import joblib
+import numpy as np
+from sklearn.metrics import classification_report, accuracy_score
+
+# Load the model
+model = tf.keras.models.load_model("dga_model.keras")
+X_test, y_test = joblib.load("test_data.pkl")
+label_encoder = joblib.load("label_encoder.pkl")
+tokenizer = joblib.load("tokenizer.pkl")
+
+# Make predictions on the test set
+y_pred = (model.predict(X_test) > 0.5).astype("int32").flatten()
+
+# Calculate accuracy
+accuracy = accuracy_score(y_test, y_pred)
+print(f"Accuracy: {accuracy:.4f}")
+
+# Generate the classification report
+report = classification_report(y_test, y_pred, target_names=label_encoder.classes_)
+print("\nClassification Report:")
+print(report)
-- 
cgit v1.2.3