report mean fixed, datasets included

This commit is contained in:
Lorenzo Volpi 2024-04-23 16:29:19 +02:00
parent e0b80167b9
commit 522d074087
1 changed files with 3 additions and 3 deletions

View File

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