Update organize codes

This commit is contained in:
D-X-Y 2021-03-22 13:41:07 +08:00
parent bfa94d9b80
commit dd27f15e27

View File

@ -84,7 +84,9 @@ class QResult:
return head_str, value_str return head_str, value_str
def compare_results(heads, values, names, space=10, separate="& ", verbose=True, sort_key=False): def compare_results(
heads, values, names, space=10, separate="& ", verbose=True, sort_key=False
):
for idx, x in enumerate(heads): for idx, x in enumerate(heads):
assert x == heads[0], "[{:}] \n{:}\nvs\n{:}".format(idx, x, heads[0]) assert x == heads[0], "[{:}] \n{:}\nvs\n{:}".format(idx, x, heads[0])
new_head = QResult.full_str("Name", space) + separate + heads[0] new_head = QResult.full_str("Name", space) + separate + heads[0]
@ -104,7 +106,7 @@ def compare_results(heads, values, names, space=10, separate="& ", verbose=True,
else: else:
lines = info_str_dict["lines"] lines = info_str_dict["lines"]
for xline in lines: for xline in lines:
print(xline) print(xline + "\\\\")
return info_str_dict return info_str_dict
@ -160,7 +162,11 @@ def query_info(save_dir, verbose):
print("There are no valid recorders for {:}".format(experiment)) print("There are no valid recorders for {:}".format(experiment))
continue continue
else: else:
print("There are {:} valid recorders for {:}".format(len(recorders), experiment.name)) print(
"There are {:} valid recorders for {:}".format(
len(recorders), experiment.name
)
)
head_str, value_str = result.info(all_keys, verbose=verbose) head_str, value_str = result.info(all_keys, verbose=verbose)
head_strs.append(head_str) head_strs.append(head_str)
value_strs.append(value_str) value_strs.append(value_str)