Keep random subset of the test set

nouiz · nouiz · commit 6e387832af57 · 2015-02-03T09:39:03.000-05:00
diff --git a/code/lstm.py b/code/lstm.py
@@ -415,7 +415,13 @@ def train_lstm(
     train, valid, test = load_data(n_words=n_words, valid_portion=0.05,
                                    maxlen=maxlen)
     if test_size > 0:
-        test = (test[0][:test_size], test[1][:test_size])
+        # The test set is sorted by size, but we want to keep random
+        # size example.  So we must select a random selection of the
+        # examples.
+        idx = numpy.arange(len(test[0]))
+        random.shuffle(idx)
+        idx = idx[:test_size]
+        test = ([test[0][n] for n in idx], [test[1][n] for n in idx])
 
     ydim = numpy.max(train[1]) + 1
 
