Mercurial > pylearn
changeset 99:a8da709eb6a9
in ArrayDataSet.__init__ if a columns is an index, we change it to be a list that containt only this index. This way, we remove the special case where the columns is an index for all subsequent call.
This was possing trouble with numpy.vstack() called by MinibatchWrapAroundIterator.next
| author | Frederic Bastien <bastienf@iro.umontreal.ca> |
|---|---|
| date | Tue, 06 May 2008 13:57:36 -0400 |
| parents | 7186e4f502d1 |
| children | 574f4db76022 |
| files | dataset.py |
| diffstat | 1 files changed, 1 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/dataset.py Tue May 06 13:50:54 2008 -0400 +++ b/dataset.py Tue May 06 13:57:36 2008 -0400 @@ -926,6 +926,7 @@ for fieldname, fieldcolumns in self.fields_columns.items(): if type(fieldcolumns) is int: assert fieldcolumns>=0 and fieldcolumns<data_array.shape[1] + self.fields_columns[fieldname]=[fieldcolumns] elif type(fieldcolumns) is slice: start,step=None,None if not fieldcolumns.start: