Update baselines

This commit is contained in:
D-X-Y
2021-03-28 10:57:20 +00:00
parent 0055511829
commit 53cb5f1fdd
4 changed files with 24 additions and 10 deletions

View File

@@ -67,6 +67,18 @@ def extend_transformer_settings(alg2configs, name):
return alg2configs
def remove_PortAnaRecord(alg2configs):
alg2configs = copy.deepcopy(alg2configs)
for key, config in alg2configs.items():
xlist = config["task"]["record"]
new_list = []
for x in xlist:
if x["class"] != "PortAnaRecord":
new_list.append(x)
config["task"]["record"] = new_list
return alg2configs
def retrieve_configs():
# https://github.com/microsoft/qlib/blob/main/examples/benchmarks/
config_dir = (lib_dir / ".." / "configs" / "qlib").resolve()
@@ -105,6 +117,12 @@ def retrieve_configs():
)
)
alg2configs = extend_transformer_settings(alg2configs, "TSF")
alg2configs = remove_PortAnaRecord(alg2configs)
print(
"There are {:} algorithms : {:}".format(
len(alg2configs), list(alg2configs.keys())
)
)
return alg2configs

View File

@@ -223,7 +223,7 @@ if __name__ == "__main__":
info_dict["heads"],
info_dict["values"],
info_dict["names"],
space=12,
space=14,
verbose=True,
sort_key=True,
)