IUM_06

.gitignore

@@ -3,3 +3,4 @@ creditcard.csv
 data
 model/model.keras
 stats_data
+evaluation

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")