Skip to content

关于上证50数据 #6

Description

@msg14

您好,我想用上证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")

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions