(pdir, ndir)
| 31 | return train_data, labels |
| 32 | |
| 33 | def get_dataset(pdir, ndir): |
| 34 | train_data = [] |
| 35 | labels = [] |
| 36 | train_data, labels = get_data(train_data, labels, pdir, lableType=1) |
| 37 | train_data, labels = get_data(train_data, labels, ndir, lableType=-1) |
| 38 | |
| 39 | return np.array(train_data, dtype=np.float32), np.array(labels, dtype=np.int32) |
| 40 | |
| 41 | if __name__ == '__main__': |
| 42 | a, b = get_dataset("pdir/", "ndir/") |