def _csv(model: torch.nn.Module, agg: StatsAggregator, model_name: str, mode: Datasets) -> str:
params = sum([np.prod(p.size()) for p in model.parameters()])
return ",".join(map(str, [model_name,
mode.value,
params,
agg.accuracy(),
*map(agg.precision, agg.classes),
*map(agg.recall, agg.classes),
*map(agg.f1_score, agg.classes),
*agg.confusion_matrix.flatten()
]))
def test_accuracy(aggregator: StatsAggregator):
assert np.isclose(aggregator.accuracy(), 5 / 8)