| 35 | print('[{}]{}'.format(t, s)) |
| 36 | |
| 37 | def load_data(filepath): |
| 38 | with gzip.open(filepath, 'rb') as f: |
| 39 | train_set, valid_set, test_set = pickle.load(f, encoding='bytes') |
| 40 | traindata = train_set[0].astype(np.float32) |
| 41 | validdata = valid_set[0].astype(np.float32) |
| 42 | testdata = test_set[0].astype(np.float32) |
| 43 | trainlabel = train_set[1].astype(np.float32) |
| 44 | validlabel = valid_set[1].astype(np.float32) |
| 45 | testlabel = test_set[1].astype(np.float32) |
| 46 | return traindata, trainlabel, validdata, validlabel, testdata, testlabel |
| 47 | |
| 48 | def download_data(gzfile, url): |
| 49 | if os.path.exists(gzfile): |