Build fix for the new syntax in the score logging.

aigamedev · Apr 10, 2015 · cf4e14a · cf4e14a
1 parent 3c8585e
commit cf4e14a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README.rst b/README.rst
@@ -37,6 +37,8 @@ Benchmarks
 
 Here are the results of testing 10 epochs of training for two-thirds of the original MNIST data, on Ubuntu 14.04 and a GeForce GTX 650 (Memory: 1024Mb, Cores: 384).  You can run ``examples/bench_mnist.py`` to get the results.
 
+.. class:: center
+
 ==========  ============  ===============  ===================
    MNIST      sknn.mlp      nolearn.dbn      nolearn.lasagne
 ==========  ============  ===============  ===================

diff --git a/sknn/tests/test_training.py b/sknn/tests/test_training.py
@@ -34,10 +34,10 @@ def test_VerboseRegressor(self):
         nn = sknn.mlp.MultiLayerPerceptronRegressor(layers=[("Linear",)], verbose=1, n_iter=1)
         a_in, a_out = numpy.zeros((8,16)), numpy.zeros((8,4))
         nn.fit(a_in, a_out)
-        assert_in("Epoch 0, R^2 1.000000", self.buf.getvalue())
+        assert_in("Epoch 0, score = 1.000000", self.buf.getvalue())
 
     def test_VerboseClassifier(self):
         nn = sknn.mlp.MultiLayerPerceptronClassifier(layers=[("Linear",)], verbose=1, n_iter=1)
         a_in, a_out = numpy.zeros((8,16)), numpy.zeros((8,1), dtype=numpy.int32)
         nn.fit(a_in, a_out)
-        assert_in("Epoch 0, R^2 1.000000", self.buf.getvalue())
+        assert_in("Epoch 0, score = 1.000000", self.buf.getvalue())