Shuqi-li / PEN

The code of PEN: Prediction-Explanation Network to Forecast Stock Price Movement with Better Explainability
16 stars 0 forks source link

代码疑问 #1

Open 1qingqing1 opened 1 year ago

1qingqing1 commented 1 year ago

您好,请问我在运行的时候,在DataPipe.py的index_token函数的indexed_token_dict[token_list_cp[id]] = id这一行执行的时候,报Keyerror错误,然后我发现token_list_cp此时是一个字典类型的变量,存储的应该是按类别划分的股票名称: {'materials': ['XOM', 'RDS-B', 'PTR', 'CVX', 'TOT', 'BP', 'BHP', 'SNP', 'SLB', 'BBL'], 'consumer_goods': ['AAPL', 'PG', 'BUD', 'KO', 'PM', 'TM', 'PEP', 'UN', 'UL', 'MO'], 'healthcare': ['JNJ', 'PFE', 'NVS', 'UNH', 'MRK', 'AMGN', 'MDT', 'ABBV', 'SNY', 'CELG'], 'services': ['AMZN', 'BABA', 'WMT', 'CMCSA', 'HD', 'DIS', 'MCD', 'CHTR', 'UPS', 'PCLN'], 'utilities': ['NEE', 'DUK', 'D', 'SO', 'NGG', 'AEP', 'PCG', 'EXC', 'SRE', 'PPL'], 'cong': ['IEP', 'HRG', 'CODI', 'REX', 'SPLP', 'PICO', 'AGFS', 'GMRE'], 'finance': ['BCH', 'BSAC', 'BRK-A', 'JPM', 'WFC', 'BAC', 'V', 'C', 'HSBC', 'MA'], 'industrial_goods': ['GE', 'MMM', 'BA', 'HON', 'UTX', 'LMT', 'CAT', 'GD', 'DHR', 'ABB'], 'tech': ['GOOG', 'MSFT', 'FB', 'T', 'CHL', 'ORCL', 'TSM', 'VZ', 'INTC', 'CSCO']} 请问该如何解决呢?

Sia2010 commented 11 months ago

请问代码里的vocab.txt是什么啊