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