crslab.model.utils package

Subpackages

• crslab.model.utils.modules package

Submodules

crslab.model.utils.functions.edge_to_pyg_format(edge, type='RGCN')

crslab.model.utils.functions.sort_for_packed_sequence(lengths: torch.Tensor)

Parameters

lengths – 1D array of lengths

Returns

sorted_lengths (lengths in descending order), sorted_idx (indices to sort), rev_idx (indices to retrieve original order)

Module contents