Commit 8428d970 authored by Naundorf's avatar Naundorf
Browse files

Update KI-KNN

parent 940cdc07
Pipeline #6558 failed with stage
in 2 minutes and 20 seconds
......@@ -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))
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment