diff --git a/KI-KNN b/KI-KNN index 4e7bfcee13db77cbd5646699904ad0f360e3a366..d5c0e1110c3a22af87a3f8863f179fdc1096145e 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)) + + +