generate clf data for multitask pipe

multiclass_generator.py

class MulticlassGenerator:
    def __init__(self, data_tuple, batch_size=256):
        self._data = data_tuple
        self._idx = np.arange(len(data_tuple[-1]))
        self.generator = self.generate_batch(batch_size)

    def generate_batch(self, size):
        while True:
            px_ids = np.random.choice(self._idx, size, replace=False)
            samples = [p[px_ids] for p in self._data[:-1]]
            labels = self._data[-1][px_ids]
            
            yield samples+[labels], [1]*size