Update KI-KNN

8428d970 · Naundorf · 940cdc07 · 8428d970
Commit 8428d970 authored Jun 18, 2022 by Naundorf
--- 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))
+
+
+