Function _sample_mask

python/dgl/data/citation_graph.py:333–337

Create mask.

(idx, l)

def _sample_mask(idx, l):
    """Create mask."""
    mask = np.zeros(l)
    mask[idx] = 1
    return mask


class CoraGraphDataset(CitationGraphDataset):