aa805170e320ffdab392758c230d5c7078d3b75b,examples/tox21/predictor.py,InferenceLoop,customized_inference,#InferenceLoop#Any#Any#Any#,78

Before Change

        for batch in iterator:
            x = converter(batch, device=device)
            y = self.predictor.predict(*x)
            ret.append(cuda.to_cpu(y.data))
        return np.concatenate(ret, axis=0)

    def inference(self, X):

After Change

        for batch in iterator:
            x = converter(batch, device=device)
            y_prob = self.predictor.predict(*x)
            y_prob = cuda.to_cpu(y_prob.data)
            y_pred = np.where(y_prob > .5, 1, -1)
            ret.append(y_pred)
        return np.concatenate(ret, axis=0)

    def inference(self, X):

Instances