Function batch_data

mnl/utils.py:222–236

Split data into batches. Args: data: List of data items batch_size: Size of each batch Returns: List of batches

(data: List[Any], batch_size: int)

def batch_data(data: List[Any], batch_size: int) -> List[List[Any]]:
    """
    Split data into batches.
    
    Args:
        data: List of data items
        batch_size: Size of each batch
        
    Returns:
        List of batches
    """
    batches = []
    for i in range(0, len(data), batch_size):
        batches.append(data[i:i + batch_size])
    return batches