| 3 | import pdb |
| 4 | |
| 5 | class MarginSampling(Strategy): |
| 6 | def __init__(self, X, Y, X_te, Y_te, idxs_lb, net, handler, args): |
| 7 | super(MarginSampling, self).__init__(X, Y, X_te, Y_te, idxs_lb, net, handler, args) |
| 8 | |
| 9 | def query(self, n): |
| 10 | idxs_unlabeled = np.arange(self.n_pool)[~self.idxs_lb] |
| 11 | probs = self.predict_prob(self.X[idxs_unlabeled], self.Y.numpy()[idxs_unlabeled]) |
| 12 | probs_sorted, idxs = probs.sort(descending=True) |
| 13 | U = probs_sorted[:, 0] - probs_sorted[:,1] |
| 14 | return idxs_unlabeled[U.sort()[1].numpy()[:n]] |
nothing calls this directly
no outgoing calls
no test coverage detected