From 2ed6ff22edd7af94d1603a3f9f4423f950317a5a Mon Sep 17 00:00:00 2001 From: Hu Zetian <50768559+Gikiman@users.noreply.github.com> Date: Fri, 16 Jun 2023 09:46:41 +0800 Subject: [PATCH] Update encoder.py --- agent/SARL/encoder/encoder.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/agent/SARL/encoder/encoder.py b/agent/SARL/encoder/encoder.py index fba2eb53..6b17df6c 100644 --- a/agent/SARL/encoder/encoder.py +++ b/agent/SARL/encoder/encoder.py @@ -186,9 +186,9 @@ def __init__(self, args): self.test_label_list, self.test_df_list = prepart_m_lstm_data( self.test_data, self.num_day, self.technical_indicator) self.train_dataset = m_lstm_dataset(self.train_df_list, - self.train_label_list) + self.train_label_list,self.num_day) self.valid_dataset = m_lstm_dataset(self.valid_df_list, - self.valid_label_list) + self.valid_label_list,self.num_day) train_dataloader = DataLoader(self.train_dataset, batch_size=self.batch_size, shuffle=True) @@ -219,4 +219,4 @@ def set_seed(self): if __name__ == "__main__": args = parser.parse_args() - a = encoder(args) \ No newline at end of file + a = encoder(args)