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

    accuracy = accuracy_score(y_test, y_pred)
    precision_micro = precision_score(y_test, y_pred, average="micro")
    recall_micro = recall_score(y_test, y_pred, average="micro")

    with open(r"evaluation/metrics.txt", "a") as f:
        f.write(f"Accuracy: {accuracy}\n")
        f.write(f"Micro-average Precision: {precision_micro}\n")
        f.write(f"Micro-average Recall: {recall_micro}\n")
        f.write(f"\n")


if __name__ == "__main__":
    main()