From f5d903caadb00b3e2f68c74cf9da7a19cf4545f7 Mon Sep 17 00:00:00 2001 From: Luca Deri Date: Sat, 26 Oct 2024 21:15:36 +0200 Subject: Moved new DGA code --- dga/scikit-learn_tests/test_script.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 dga/scikit-learn_tests/test_script.py (limited to 'dga/scikit-learn_tests/test_script.py') diff --git a/dga/scikit-learn_tests/test_script.py b/dga/scikit-learn_tests/test_script.py new file mode 100644 index 000000000..4ded249f8 --- /dev/null +++ b/dga/scikit-learn_tests/test_script.py @@ -0,0 +1,23 @@ +import joblib +from sklearn.neural_network import MLPClassifier +from sklearn.metrics import classification_report, accuracy_score +import time + +mlp = joblib.load('mlp_model.joblib') +X_test = joblib.load('X_test.joblib') +y_test = joblib.load('y_test.joblib') +label_encoder = joblib.load('label_encoder.joblib') + +# Perform prediction +start = time.time() +y_pred = mlp.predict(X_test) +print(f"Prediction time: {time.time()-start:.2f} seconds") + +# Evaluate the model +accuracy = accuracy_score(y_test, y_pred) +report = classification_report(y_test, y_pred, target_names=label_encoder.classes_) + +# Print the results +print(f"Accuracy: {accuracy:.4f}") +print("\nClassification Report:") +print(report) -- cgit v1.2.3