File tree Expand file tree Collapse file tree 2 files changed +22
-2
lines changed Expand file tree Collapse file tree 2 files changed +22
-2
lines changed Original file line number Diff line number Diff line change @@ -326,7 +326,8 @@ def main(param):
326
326
'best test F1' , param ['tf1' ],
327
327
'with the model' , folder )
328
328
329
- def test_rnnslu (n_epochs ):
329
+
330
+ if __name__ == '__main__' :
330
331
# best model
331
332
s = {'fold' : 3 ,
332
333
# 5 folds 0,1,2,3,4
Original file line number Diff line number Diff line change 15
15
16
16
17
17
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 )
19
38
20
39
21
40
def test_logistic_sgd ():
You can’t perform that action at this time.
0 commit comments