diff baseline/log_reg/log_reg.py @ 199:777f48ba30df

Add MSE cost to log_reg.py
author Arnaud Bergeron <abergeron@gmail.com>
date Tue, 02 Mar 2010 18:43:54 -0500
parents 5d88ed99c0af
children 7be1f086a89e
line wrap: on
line diff
--- a/baseline/log_reg/log_reg.py	Tue Mar 02 18:16:49 2010 -0500
+++ b/baseline/log_reg/log_reg.py	Tue Mar 02 18:43:54 2010 -0500
@@ -112,6 +112,8 @@
         # i.e., the mean log-likelihood across the minibatch.
         return -T.mean( T.log( self.p_y_given_x )[ T.arange( y.shape[0] ), y ] )
 
+    def MSE(self, y):
+        return -T.mean(abs((self.p_t_given_x)[T.arange(y.shape[0]), y]-y)**2)
 
     def errors( self, y ):
         """Return a float representing the number of errors in the minibatch