鸽子 1 year ago
parent c1b927d7d4
commit 3794fcb378

@ -7,6 +7,7 @@ import os
DAYS_FOR_TRAIN = 9
class LSTM_Regression(nn.Module):
def __init__(self, input_size, hidden_size, output_size=1, num_layers=2):

Loading…
Cancel
Save