(pdir, ndir)
| 30 | return train_data, labels |
| 31 | |
| 32 | def get_dataset(pdir, ndir): |
| 33 | train_data = [] |
| 34 | labels = [] |
| 35 | train_data, labels = get_data(train_data, labels, pdir, lableType=1) |
| 36 | train_data, labels = get_data(train_data, labels, ndir, lableType=-1) |
| 37 | |
| 38 | return np.array(train_data, dtype=np.float32), np.array(labels, dtype=np.int32) |
| 39 | |
| 40 | |
| 41 | def svm_train(positive_dir, negative_dir): |