Update model.py
Browse files
model.py
CHANGED
@@ -2,7 +2,7 @@ import torch.nn as nn
|
|
2 |
import torch
|
3 |
|
4 |
class LSTMClassifier(nn.Module):
|
5 |
-
def __init__(self, input_size=1, hidden_size=
|
6 |
bidirectional=True, dropout=0.0, num_classes=2):
|
7 |
super(LSTMClassifier, self).__init__()
|
8 |
self.hidden_size = hidden_size
|
|
|
2 |
import torch
|
3 |
|
4 |
class LSTMClassifier(nn.Module):
|
5 |
+
def __init__(self, input_size=1, hidden_size=64, num_layers=1,
|
6 |
bidirectional=True, dropout=0.0, num_classes=2):
|
7 |
super(LSTMClassifier, self).__init__()
|
8 |
self.hidden_size = hidden_size
|