(file_path)
| 34 | |
| 35 | |
| 36 | def load_train_data(file_path): |
| 37 | f = gzip.open(file_path, 'rb') |
| 38 | if sys.version_info.major > 2: |
| 39 | train_set, valid_set, test_set = pickle.load(f, encoding='latin1') |
| 40 | else: |
| 41 | train_set, valid_set, test_set = pickle.load(f) |
| 42 | traindata = train_set[0].astype(np.float32) |
| 43 | validdata = valid_set[0].astype(np.float32) |
| 44 | print(traindata.shape, validdata.shape) |
| 45 | return traindata, validdata |
| 46 | |
| 47 | |
| 48 | def train(data_file, use_gpu, num_epoch=10, batch_size=100): |