diff --git a/pyproject.toml b/pyproject.toml index 35af11b..f594632 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "gasbench" -version = "0.9.2" +version = "0.9.3" description = "GASBench - ML model benchmark evaluation package" readme = "README.md" requires-python = ">=3.10" diff --git a/src/gasbench/benchmarks/recording.py b/src/gasbench/benchmarks/recording.py index 5612472..7fd3026 100644 --- a/src/gasbench/benchmarks/recording.py +++ b/src/gasbench/benchmarks/recording.py @@ -646,6 +646,8 @@ def _compute_aug_metrics( "aug_binary_mcc": aug_metrics.calculate_binary_mcc(), "aug_binary_ce": aug_metrics.calculate_binary_cross_entropy(), "aug_binary_brier": aug_metrics.calculate_brier(), + "aug_gorodkin_mcc": aug_metrics.calculate_multiclass_mcc(), + "aug_multiclass_brier": aug_metrics.calculate_multiclass_brier(), } # Per-sample degradation: join augmented rows to their base counterpart via sample_id