changeset 175:e9a95e19e6f8

Added a Print Op
author Yoshua Bengio <bengioy@iro.umontreal.ca>
date Tue, 13 May 2008 15:11:24 -0400
parents fb4837eed1a6
children 6ee54861134e
files misc.py mlp.py
diffstat 2 files changed, 23 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/misc.py	Tue May 13 14:28:47 2008 -0400
+++ b/misc.py	Tue May 13 15:11:24 2008 -0400
@@ -1,3 +1,24 @@
+
+import theano
+
+class Print(theano.Op):
+    def __init__(self,message=""):
+        self.message=message
+        self.view_map={0:[0]}
+
+    def make_node(self,xin):
+        xout = xin.type.make_result()
+        return theano.Apply(op = self, inputs = [xin], outputs=[xout])
+
+    def perform(self,node,inputs,output_storage):
+        xin, = inputs
+        xout, = output_storage
+        xout[0] = xin
+        print self.message,xin
+
+    def grad(self,input,output_gradients):
+        return output_gradients
+
 
 def unique_elements_list_intersection(list1,list2):
     """
--- a/mlp.py	Tue May 13 14:28:47 2008 -0400
+++ b/mlp.py	Tue May 13 15:11:24 2008 -0400
@@ -9,7 +9,7 @@
 from theano import tensor as t
 from nnet_ops import *
 import math
-
+from misc import *
 
 class OneHiddenLayerNNetClassifier(OnlineGradientTLearner):
     """
@@ -88,7 +88,7 @@
         self._b2 = t.row('b2')
         self._regularization_term = self._L2_regularizer * (t.sum(self._W1*self._W1) + t.sum(self._W2*self._W2))
         self._output_activations =self._b2+t.dot(t.tanh(self._b1+t.dot(self._input,self._W1.T)),self._W2.T)
-        self._nll,self._output = crossentropy_softmax_1hot(self._output_activations,self._target_vector)
+        self._nll,self._output = crossentropy_softmax_1hot(Print("output_activations")(self._output_activations),self._target_vector)
         self._output_class = t.argmax(self._output,1)
         self._class_error = t.neq(self._output_class,self._target_vector)
         self._minibatch_criterion = self._nll + self._regularization_term / t.shape(self._input)[0]