diff --git a/run.py b/run.py index 5d2a934..f05795f 100644 --- a/run.py +++ b/run.py @@ -12,8 +12,8 @@ # 初始化模型和数据 def initialize_model_and_data(): INPUT_SIZE = 3 - HIDDEN_SIZE = 10 - NUM_LAYERS = 1 + HIDDEN_SIZE = 32 + NUM_LAYERS = 3 OUTPUT_SIZE = 1 lstm = LSTM(INPUT_SIZE, HIDDEN_SIZE, NUM_LAYERS, OUTPUT_SIZE)