Mercurial > pylearn
changeset 1403:6ade5b39b773
int8 should be enough to represent digits from 0 to 9
| author | Pascal Lamblin <lamblinp@iro.umontreal.ca> |
|---|---|
| date | Fri, 21 Jan 2011 20:40:57 -0500 |
| parents | b14f3d6f5cd4 |
| children | 89017617ab36 |
| files | pylearn/datasets/MNIST.py |
| diffstat | 1 files changed, 2 insertions(+), 2 deletions(-) [+] |
line wrap: on
line diff
--- a/pylearn/datasets/MNIST.py Fri Jan 21 17:05:46 2011 -0500 +++ b/pylearn/datasets/MNIST.py Fri Jan 21 20:40:57 2011 -0500 @@ -15,7 +15,7 @@ Returns two matrices: x, y. x has N rows of 784 columns. Each row of x represents the 28x28 grey-scale pixels in raster order. y is a vector of N integers. Each element y[i] is the label of the i'th row of x. - + """ if path is None: path = os.path.join(data_root(), 'mnist','mnist_all.pmat') @@ -24,7 +24,7 @@ rows=dat.getRows(0,n) - return rows[:,0:-1], numpy.asarray(rows[:,-1], dtype='int64') + return rows[:,0:-1], numpy.asarray(rows[:,-1], dtype='int8') #What is the purpose of this fct?