Add ensemble test case.

Author: riga

tests/core.py

@@ -60,3 +60,34 @@ def test_eval(self):
         # no element in the diff array should be larger than 1e-7
         maxdiff = np.max(np.abs(rtf - rtd))
         self.assertLess(maxdiff, 1e-7)
+
+    def test_ensemble_eval(self):
+        simple_model2 = td.Model()
+        y2, sess2 = self.get("simple2", "y", "sess")
+        simple_model2.add(y2, tf_sess=sess2)
+
+        simple_model2.get("input_1").name = "input:0"
+        simple_model2.get("output_1").name = "output:0"
+        simple_model2.get("keep_prob_1").name = "keep_prob:0"
+
+        simple_ensemble = td.Ensemble()
+        simple_ensemble.models = [self.simple_model, simple_model2]
+
+        inp, outp, kp = simple_ensemble.get("input", "output", "keep_prob")
+
+        # create an input batch
+        examples = np.random.rand(1000, 10).astype("float32")
+
+        # eval both models manually and build the mean
+        x1, y1, keep_prob1 = self.simple_model.get("input", "output", "keep_prob")
+        r1 = y1.eval({x1: examples, keep_prob1: 1.0})
+        x2, y2, keep_prob2 = simple_model2.get("input", "output", "keep_prob")
+        r2 = y2.eval({x2: examples, keep_prob2: 1.0})
+        rm = np.add(r1, r2) / 2.
+
+        # then, eval the ensemble
+        re = outp.eval({inp: examples, kp: 1.0})
+
+        # no element in the diff array should be larger than 1e-7
+        maxdiff = np.max(np.abs(re - rm))
+        self.assertLess(maxdiff, 1e-7)

tests/models/simple2.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+
+
+import tensorflow as tf
+
+
+sess = tf.Session()
+
+x = tf.placeholder(tf.float32, shape=[None, 10], name="input")
+keep_prob = tf.placeholder(tf.float32, name="keep_prob")
+
+W = tf.Variable(tf.truncated_normal([10, 5], stddev=0.05))
+b = tf.Variable(tf.zeros([5]))
+
+W_drop = tf.nn.dropout(W, keep_prob)
+
+y = tf.nn.softmax(tf.matmul(x, W_drop) + b, name="output")
+
+sess.run(tf.initialize_all_variables())