def __next__(self):
    if (self.index >= self.length):
        raise StopIteration
    self.index += 1
    return (self.predict, self.label)
def __next__(self):
    if (self.index >= self.length):
        raise StopIteration
    self.index += 1
    return (self.predict, self.label)