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):
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
Original file line number Diff line number Diff line change 1515
1616
1717def 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
2140def test_logistic_sgd ():
You can’t perform that action at this time.
0 commit comments