Python StatsAggregator.accuracy Examples

Example #1

File: evaluation.py Project: kapitsa2811/chesscog

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()
                              ]))

Example #2

File: test_statistics.py Project: kapitsa2811/chesscog

def test_accuracy(aggregator: StatsAggregator):
    assert np.isclose(aggregator.accuracy(), 5 / 8)