您好,我想用上证50跑个demo,您给的example_dataset里面好像缺少了一些数据文件(main.py代码中的{stock_pool_name}.parq和all_ashare_label.parq),能否上传一下。
ref:
stock_pool_name = args.stock_pool
assert isinstance(stock_pool_name, str)
stock_pool = pd.read_parquet(f"{ROOT_PATH}/stock_prediction_benchmark/stock_disagreement/dataset/{stock_pool_name}.parq")
industry = pd.read_parquet("{ROOT_PATH}/stock_prediction_benchmark/stock_disagreement/dataset/stock_basic_data.parq")
stock_pool = stock_pool.merge(industry[["Stock", "Name", "Industry"]], on=["Stock"], how="left")
Change to your label file url here.
stock_labels = pd.read_parquet("{ROOT_PATH}/stock_prediction_benchmark/stock_disagreement/dataset/all_ashare_label.parq")
您好,我想用上证50跑个demo,您给的example_dataset里面好像缺少了一些数据文件(main.py代码中的{stock_pool_name}.parq和all_ashare_label.parq),能否上传一下。
ref:
stock_pool_name = args.stock_pool
assert isinstance(stock_pool_name, str)
stock_pool = pd.read_parquet(f"{ROOT_PATH}/stock_prediction_benchmark/stock_disagreement/dataset/{stock_pool_name}.parq")
industry = pd.read_parquet("{ROOT_PATH}/stock_prediction_benchmark/stock_disagreement/dataset/stock_basic_data.parq")
stock_pool = stock_pool.merge(industry[["Stock", "Name", "Industry"]], on=["Stock"], how="left")
Change to your label file url here.
stock_labels = pd.read_parquet("{ROOT_PATH}/stock_prediction_benchmark/stock_disagreement/dataset/all_ashare_label.parq")