Skip to content

Commit e1c1839

Browse files
author
Grégoire
committed
new fix
1 parent f9c425b commit e1c1839

File tree

2 files changed

+22
-2
lines changed

2 files changed

+22
-2
lines changed

code/rnnslu.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,8 @@ def main(param):
326326
'best test F1', param['tf1'],
327327
'with the model', folder)
328328

329-
def test_rnnslu(n_epochs):
329+
330+
if __name__ == '__main__':
330331
# best model
331332
s = {'fold': 3,
332333
# 5 folds 0,1,2,3,4

code/test.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,26 @@
1515

1616

1717
def test_rnnslu():
18-
rnnslu.test_rnnslu(n_epochs=1)
18+
# best model
19+
s = {'fold': 3,
20+
# 5 folds 0,1,2,3,4
21+
'data': 'atis',
22+
'lr': 0.0970806646812754,
23+
'verbose': 1,
24+
'decay': True,
25+
# decay on the learning rate if improvement stops
26+
'win': 7,
27+
# number of words in the context window
28+
'nhidden': 200,
29+
# number of hidden units
30+
'seed': 345,
31+
'emb_dimension': 50,
32+
# dimension of word embedding
33+
'nepochs': 1,
34+
# 60 is recommended
35+
'savemodel': True}
36+
37+
rnnslu.main(s)
1938

2039

2140
def test_logistic_sgd():

0 commit comments

Comments
 (0)