We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b3b1783 commit 1a15292Copy full SHA for 1a15292
1 file changed
code/test.py
@@ -15,10 +15,6 @@
15
import lstm
16
17
18
-def test_rnnslu():
19
- rnnslu.main()
20
-
21
22
def test_logistic_sgd():
23
logistic_sgd.sgd_optimization_mnist(n_epochs=10)
24
@@ -62,6 +58,26 @@ def test_rnnrbm():
62
58
rnnrbm.test_rnnrbm(num_epochs=1)
63
59
64
60
61
+def test_rnnslu():
+ s = {'fold': 3,
+ # 5 folds 0,1,2,3,4
+ '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
81
def test_lstm():
82
lstm.train_lstm(max_epochs=1, test_size=1000, saveto='')
83
0 commit comments