From 522d0740875cb3fdbbccc7f0b037f9c21cb5d659 Mon Sep 17 00:00:00 2001 From: Lorenzo Volpi Date: Tue, 23 Apr 2024 16:29:19 +0200 Subject: [PATCH] report mean fixed, datasets included --- examples/ucimulti_experiments.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/ucimulti_experiments.py b/examples/ucimulti_experiments.py index b01163a..16144cd 100644 --- a/examples/ucimulti_experiments.py +++ b/examples/ucimulti_experiments.py @@ -62,9 +62,9 @@ if __name__ == '__main__': with open(global_result_path + '.csv', 'at') as csv: - for dataset in qp.datasets.UCI_MULTICLASS_DATASETS[:5]: + for dataset in qp.datasets.UCI_MULTICLASS_DATASETS[:12]: - if dataset in ['covertype', 'diabetes']: + if dataset in []: continue print('init', dataset) @@ -106,7 +106,7 @@ if __name__ == '__main__': ) report.to_csv(local_result_path) - means = report.mean() + means = report.mean(numeric_only=True) csv.write(f'{method_name}\t{dataset}\t{means["mae"]:.5f}\t{means["mrae"]:.5f}\n') csv.flush()