From a8cf8d2829f9c1a9d3d80c5d01d7ac14faeee197 Mon Sep 17 00:00:00 2001 From: Mateusz Date: Sat, 4 May 2024 15:42:16 +0200 Subject: [PATCH] IUM_06 --- .gitignore | 3 ++- metrics.py | 5 +---- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/.gitignore b/.gitignore index 06ccd0b..cf51aef 100644 --- a/.gitignore +++ b/.gitignore @@ -2,4 +2,5 @@ creditcardfraud.zip creditcard.csv data model/model.keras -stats_data \ No newline at end of file +stats_data +evaluation \ No newline at end of file diff --git a/metrics.py b/metrics.py index 2a14e17..9f5a041 100644 --- a/metrics.py +++ b/metrics.py @@ -2,16 +2,13 @@ from sklearn.metrics import ( accuracy_score, precision_score, recall_score, - f1_score, - mean_squared_error, ) -import numpy as np import pandas as pd def main(): y_test = pd.read_csv("data/y_test.csv") - y_pred = pd.read_csv("evaluation/y_pred.csv") + y_pred = pd.read_csv("evaluation/y_pred.csv", header=None) accuracy = accuracy_score(y_test, y_pred) precision_micro = precision_score(y_test, y_pred, average="micro")