2024-05-04 15:54:55 +02:00
|
|
|
from sklearn.metrics import confusion_matrix
|
2024-05-04 15:25:54 +02:00
|
|
|
import pandas as pd
|
2024-05-04 16:19:51 +02:00
|
|
|
import sys
|
2024-05-04 15:25:54 +02:00
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
y_test = pd.read_csv("data/y_test.csv")
|
2024-05-04 15:42:16 +02:00
|
|
|
y_pred = pd.read_csv("evaluation/y_pred.csv", header=None)
|
2024-05-04 16:19:51 +02:00
|
|
|
build_number = sys.argv[1]
|
2024-05-04 15:25:54 +02:00
|
|
|
|
2024-05-04 15:54:55 +02:00
|
|
|
cm = confusion_matrix(y_test, y_pred)
|
|
|
|
accuracy = cm[1, 1] / (cm[1, 0] + cm[1, 1])
|
2024-05-04 15:25:54 +02:00
|
|
|
|
|
|
|
with open(r"evaluation/metrics.txt", "a") as f:
|
2024-05-04 16:19:51 +02:00
|
|
|
f.write(f"{accuracy},{build_number}\n")
|
2024-05-04 15:25:54 +02:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
main()
|