()
| 19 | |
| 20 | |
| 21 | def getKaggleMNIST(): |
| 22 | # MNIST data: |
| 23 | # column 0 is labels |
| 24 | # column 1-785 is data, with values 0 .. 255 |
| 25 | # total size of CSV: (42000, 1, 28, 28) |
| 26 | train = pd.read_csv('../large_files/train.csv').values.astype(np.float32) |
| 27 | train = shuffle(train) |
| 28 | |
| 29 | Xtrain = train[:-1000,1:] / 255 |
| 30 | Ytrain = train[:-1000,0].astype(np.int32) |
| 31 | |
| 32 | Xtest = train[-1000:,1:] / 255 |
| 33 | Ytest = train[-1000:,0].astype(np.int32) |
| 34 | return Xtrain, Ytrain, Xtest, Ytest |
| 35 | |
| 36 | |
| 37 | def init_weights(shape): |
no outgoing calls