report mean fixed, datasets included
This commit is contained in:
parent
e0b80167b9
commit
522d074087
examples
|
@ -62,9 +62,9 @@ if __name__ == '__main__':
|
||||||
|
|
||||||
with open(global_result_path + '.csv', 'at') as csv:
|
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
|
continue
|
||||||
|
|
||||||
print('init', dataset)
|
print('init', dataset)
|
||||||
|
@ -106,7 +106,7 @@ if __name__ == '__main__':
|
||||||
)
|
)
|
||||||
report.to_csv(local_result_path)
|
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.write(f'{method_name}\t{dataset}\t{means["mae"]:.5f}\t{means["mrae"]:.5f}\n')
|
||||||
csv.flush()
|
csv.flush()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue