Mercurial > pylearn
diff dataset.py @ 351:7545207466d4
debugged RenamedFieldsDataSet
author | Frederic Bastien <bastienf@iro.umontreal.ca> |
---|---|
date | Tue, 17 Jun 2008 16:47:33 -0400 |
parents | a22ea54a19ed |
children | 47538a45b878 |
line wrap: on
line diff
--- a/dataset.py Tue Jun 17 16:20:17 2008 -0400 +++ b/dataset.py Tue Jun 17 16:47:33 2008 -0400 @@ -692,6 +692,7 @@ assert len(src_fieldnames)==len(new_fieldnames) self.valuesHStack = src.valuesHStack self.valuesVStack = src.valuesVStack + self.lookup_fields = Example(new_fieldnames,src_fieldnames) def __len__(self): return len(self.src) @@ -719,7 +720,11 @@ def minibatches_nowrap(self,fieldnames,minibatch_size,n_batches,offset): assert self.hasFields(*fieldnames) - return self.src.minibatches_nowrap(fieldnames,minibatch_size,n_batches,offset) + cursor = Example(fieldnames,[0]*len(fieldnames)) + for batch in self.src.minibatches_nowrap([self.lookup_fields[f] for f in fieldnames],minibatch_size,n_batches,offset): + cursor._values=batch._values + yield cursor + def __getitem__(self,i): # return FieldsSubsetDataSet(self.src[i],self.new_fieldnames) complete_example = self.src[i]