Update baselines

This commit is contained in:
D-X-Y
2021-03-05 13:11:26 +00:00
parent 2e0325ca63
commit 2fa358fdf6
5 changed files with 286 additions and 5 deletions

View File

@@ -104,7 +104,7 @@ def main(xargs):
# start exp to train model
with R.start(experiment_name="train_tt_model"):
with R.start(experiment_name="tt_model", uri=xargs.save_dir):
set_log_basic_config(R.get_recorder().root_uri / 'log.log')
model = init_instance_by_config(model_config)
@@ -139,8 +139,6 @@ if __name__ == "__main__":
args = parser.parse_args()
provider_uri = "~/.qlib/qlib_data/cn_data" # target_dir
exp_manager = C.exp_manager
exp_manager["kwargs"]["uri"] = "file:{:}".format(Path(args.save_dir).resolve())
qlib.init(provider_uri=provider_uri, region=REG_CN, exp_manager=exp_manager)
qlib.init(provider_uri=provider_uri, region=REG_CN)
main(args)