26 lines
698 B
Python
26 lines
698 B
Python
from sklearn.metrics import (
|
|
accuracy_score,
|
|
precision_score,
|
|
recall_score,
|
|
)
|
|
import pandas as pd
|
|
|
|
|
|
def main():
|
|
y_test = pd.read_csv("data/y_test.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")
|
|
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()
|