Function data_vstack

modAL/utils/data.py:16–41

Stack vertically sparse/dense arrays and pandas data frames. Args: blocks: Sequence of modALinput objects. Returns: New sequence of vertically stacked elements.

(blocks: Sequence[modALinput])

def data_vstack(blocks: Sequence[modALinput]) -> modALinput:
    """
    Stack vertically sparse/dense arrays and pandas data frames.

    Args:
        blocks: Sequence of modALinput objects.

    Returns:
        New sequence of vertically stacked elements.
    """
    if any([sp.issparse(b) for b in blocks]):
        return sp.vstack(blocks)
    elif isinstance(blocks[0], pd.DataFrame):
        return blocks[0].append(blocks[1:])
    elif isinstance(blocks[0], np.ndarray):
        return np.concatenate(blocks)
    elif isinstance(blocks[0], list):
        return np.concatenate(blocks).tolist()

    try:
        if torch.is_tensor(blocks[0]):
            return torch.cat(blocks)
    except:
        pass

    raise TypeError("%s datatype is not supported" % type(blocks[0]))


def data_hstack(blocks: Sequence[modALinput]) -> modALinput: