Spaces:

tenperformer
/

Matchlens-API

Paused

Matchlens-API / app /fine_tuning /evaluate.py

Rishabh Shenoy

Added files to root

ab1e1af 4 days ago

422 Bytes

"""Evaluate a trained classifier without fabricating metrics."""
from sklearn.metrics import accuracy_score, confusion_matrix, f1_score


def metrics(expected: list[int], predicted: list[int]) -> dict:
    return {
        "accuracy": accuracy_score(expected, predicted),
        "macro_f1": f1_score(expected, predicted, average="macro"),
        "confusion_matrix": confusion_matrix(expected, predicted).tolist(),
    }

	"""Evaluate a trained classifier without fabricating metrics."""
	from sklearn.metrics import accuracy_score, confusion_matrix, f1_score


	def metrics(expected: list[int], predicted: list[int]) -> dict:
	return {
	"accuracy": accuracy_score(expected, predicted),
	"macro_f1": f1_score(expected, predicted, average="macro"),
	"confusion_matrix": confusion_matrix(expected, predicted).tolist(),
	}