def collate_fn(batch_data):
    """
    Collation function to distribute data to samples
    :param batch_data: batch data
    """
    graphs = []
    mut, gexpr, met, Y = [], [], [], []
    for g, mu, gex, me, y in batch_data:
        graphs.append(g)
        mut.append(mu)
        gexpr.append(gex)
        met.append(me)
        Y.append(y)
    return graphs, mut, gexpr, met, Y