unlabeled = self.unlabeled
U_num = len(unlabeled)
US = np.zeros(U_num)
prob_matrix = self.predict_proba(X=self.X[unlabeled])
S = np.argsort(-prob_matrix, axis=1)
for i, [idx, jdx] in enumerate(S[:,[0,1]]):
US[i] = prob_matrix[i, idx] - prob_matrix[i, jdx]
如下老方法运行速度慢些
for i in range(U_num):
ord_prob = np.argsort(prob_matrix[i])
US[i] = prob_matrix[i][ord_prob[-1]] - prob_matrix[i][ord_prob[-2]]