Fixed dimensions for layer

davidmarek · Jun 20, 2012 · 6d3ef68 · 6d3ef68
1 parent 3808802
commit 6d3ef68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklearn/mlp/mlp_fast.pyx b/sklearn/mlp/mlp_fast.pyx
@@ -245,7 +245,7 @@ def sgd(np.ndarray[DTYPE_t, ndim=2] X not None,
     cdef np.ndarray[DTYPE_t, ndim=2] dx_hidden = np.empty((batch_size, n_hidden))
     cdef np.ndarray[DTYPE_t, ndim=2] x_output = np.empty((batch_size, n_outs))
     cdef np.ndarray[DTYPE_t, ndim=2] delta_o = np.empty((batch_size, n_outs))
-    cdef np.ndarray[DTYPE_t, ndim=2] dx_output = np.empty((batch_size, n_hidden))
+    cdef np.ndarray[DTYPE_t, ndim=2] dx_output = np.empty((batch_size, n_outs))
 
     if y.shape[0] != n_samples:
         raise ValueError("Shapes of X and y don't fit.")