Update baselines
This commit is contained in:
		| @@ -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 | ||||
|  | ||||
|  | ||||
|   | ||||
| @@ -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, | ||||
|     ) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user