# HG changeset patch
# User Pascal Lamblin <lamblinp@iro.umontreal.ca>
# Date 1305954190 14400
# Node ID 48efafaaf7fbe5ba4666f6d9997d12d570f1b9ad
# Parent  8c10bda4bb5fb6a451aa1e8af69103610ef09790
Add function for loading the transfer labels of utlc

diff -r 8c10bda4bb5f -r 48efafaaf7fb pylearn/datasets/utlc.py
--- a/pylearn/datasets/utlc.py	Fri May 20 16:53:00 2011 -0400
+++ b/pylearn/datasets/utlc.py	Sat May 21 01:03:10 2011 -0400
@@ -110,7 +110,7 @@
         else:
             raise Exception("This dataset don't have its normalization defined")
     if transfer:
-        transfer = load_filetensor(os.path.join(config.data_root(),"UTLC","filetensor",name+"_transfer.ft"))
+        transfer = load_ndarray_transfer(name)
         return train, valid, test, transfer
     else:
         return train, valid, test
@@ -179,6 +179,17 @@
     else:
         return train, valid, test
 
+def load_ndarray_transfer(name):
+    """
+    Load the transfer labels for the training set of data set `name`.
+
+    It will be returned in ndarray format.
+    """
+    assert name in ['avicenna','harry','rita','sylvester','terry','ule']
+    transfer = load_filetensor(os.path.join(config.data_root(), 'UTLC',
+        'filetensor', name+'_transfer.ft'))
+    return transfer
+
 def load_ndarray_label(name):
     """ Load the train,valid,test data for the dataset `name`
         and return it in ndarray format.