From 8428d9706ec9abc93d5aba85e858c79fb106564b Mon Sep 17 00:00:00 2001
From: Naundorf <81nasi1bwi@hft-stuttgart.de>
Date: Sat, 18 Jun 2022 11:21:43 +0000
Subject: [PATCH] Update KI-KNN

---
 KI-KNN | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/KI-KNN b/KI-KNN
index 4e7bfce..d5c0e11 100644
--- a/KI-KNN
+++ b/KI-KNN
@@ -116,3 +116,11 @@ predict(X_test, y_test)
 plot_model_accuracy(history)
 plot_model_loss(history)
 
+def predict(X_test, y_test):
+    print("Predictions.....")
+    predictions = np.argmax(model.predict(X_test), axis=-1)
+    target_names = [constants.0, constants.1]
+    print(classification_report(y_test, predictions, target_names=target_names))
+
+
+
-- 
GitLab