89432c6c49689c5c1fd80e80db25562b305b265f,basic/evaluator.py,TempEvaluator,compare,#Any#Any#Any#,187

Before Change

    @staticmethod
    def compare(yi, ypi, yp2i):
        for start, stop in yi:
            if tuple(start) == argmax(ypi) and tuple(stop) == argmax(yp2i):
                return True
        return False

After Change

    def compare(yi, ypi, yp2i):
        for start, stop in yi:
            aypi = argmax(ypi)
            mask = np.zeros(yp2i.shape)
            mask[aypi[0], aypi[1]:] = np.ones([yp2i.shape[1] - aypi[1]])
            if tuple(start) == aypi and (stop[0], stop[1]-1) == argmax(yp2i * mask):
                return True
        return False

Instances