File tree Expand file tree Collapse file tree 1 file changed +20
-4
lines changed
Expand file tree Collapse file tree 1 file changed +20
-4
lines changed Original file line number Diff line number Diff line change 1515import lstm
1616
1717
18- def test_rnnslu ():
19- rnnslu .main ()
20-
21-
2218def test_logistic_sgd ():
2319 logistic_sgd .sgd_optimization_mnist (n_epochs = 10 )
2420
@@ -62,6 +58,26 @@ def test_rnnrbm():
6258 rnnrbm .test_rnnrbm (num_epochs = 1 )
6359
6460
61+ def test_rnnslu ():
62+ s = {'fold' : 3 ,
63+ # 5 folds 0,1,2,3,4
64+ 'data' : 'atis' ,
65+ 'lr' : 0.0970806646812754 ,
66+ 'verbose' : 1 ,
67+ 'decay' : True ,
68+ # decay on the learning rate if improvement stops
69+ 'win' : 7 ,
70+ # number of words in the context window
71+ 'nhidden' : 200 ,
72+ # number of hidden units
73+ 'seed' : 345 ,
74+ 'emb_dimension' : 50 ,
75+ # dimension of word embedding
76+ 'nepochs' : 1 , # CHANGED
77+ 'savemodel' : False }
78+ rnnslu .main (s )
79+
80+
6581def test_lstm ():
6682 lstm .train_lstm (max_epochs = 1 , test_size = 1000 , saveto = '' )
6783
You can’t perform that action at this time.
0 commit comments