Skip to content

Commit 65591aa

Browse files
nickdrhodesalexmirrington
authored andcommitted
small change to lstm argument default
1 parent 4faf8e9 commit 65591aa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code/algorithm/modules/lstm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
class BiLSTM(nn.Module):
77
"""BiLSTM which predicts multiple labels from caption data."""
88

9-
def __init__(self, classes, threshold, embeddings_dim=100,
9+
def __init__(self, classes, threshold=0.5, embeddings_dim=100,
1010
hidden_size=128):
1111
"""Initialise the BiLSTM.
1212

0 commit comments

Comments
 (0)