Spaces:
Sleeping
Sleeping
fix typo
Browse files- predict_stock_prices.py +1 -1
predict_stock_prices.py
CHANGED
|
@@ -12,7 +12,7 @@ import os
|
|
| 12 |
from typing import List, Dict, Any, Union, Tuple
|
| 13 |
|
| 14 |
class BiLSTMModel(nn.Module):
|
| 15 |
-
def
|
| 16 |
super(BiLSTMModel, self).__init__()
|
| 17 |
self.hidden_size = hidden_size
|
| 18 |
self.num_layers = num_layers
|
|
|
|
| 12 |
from typing import List, Dict, Any, Union, Tuple
|
| 13 |
|
| 14 |
class BiLSTMModel(nn.Module):
|
| 15 |
+
def __init__(self, input_size=1, hidden_size=64, num_layers=2, output_size=1):
|
| 16 |
super(BiLSTMModel, self).__init__()
|
| 17 |
self.hidden_size = hidden_size
|
| 18 |
self.num_layers = num_layers
|