ium_464913/metrics.py

from sklearn.metrics import confusion_matrix
import pandas as pd
import sys


def main():
    y_test = pd.read_csv("data/y_test.csv")
    y_pred = pd.read_csv("evaluation/y_pred.csv", header=None)
    build_number = sys.argv[1]

    cm = confusion_matrix(y_test, y_pred)
    print(
        "Recall metric in the testing dataset: ",
        cm[1, 1] / (cm[1, 0] + cm[1, 1]),
    )
    accuracy = cm[1, 1] / (cm[1, 0] + cm[1, 1])

    with open(r"evaluation/metrics.txt", "a") as f:
        f.write(f"{accuracy},{build_number}\n")


if __name__ == "__main__":
    main()