e81142f0869321cd8792e13bfc4f6fb85d9f1efc,libact/query_strategies/uncertainty_sampling.py,UncertaintySampling,make_query,#UncertaintySampling#,62

Before Change

                    ask_id = j

        else:
            raise ValueError(
                "Invalid method "%s" (available choices: ("lc", "sm", "le")"
                % self.method
                )

        return unlabeled_entry_ids[ask_id]

    def get_model(self):

After Change

        elif self.method == "sm":  // smallest margin
            dvalue = self.model.predict_real(X_pool)
            if np.shape(dvalue)[1] == 2:
                ind = [0, 1]
            else:
                // Find 2 largest decision values
                ind = np.partition(-np.abs(dvalue), 2, axis=1)[:2]
            margin = np.abs(np.abs(dvalue[:, ind[0]]) - np.abs(dvalue[:, ind[1]]))
            ask_id = np.argmin(margin)

        return unlabeled_entry_ids[ask_id]

Instances