Bering.models._sample_edges

Bering.models._sample_edges(data, num_pos_edges, num_neg_edges, num_edges_perG=10000)[source]

Sample random positive & negative edges as the input of decoder.

Parameters:

  • data (DataLoader) – Input data loader; Node coordinates and cell ids are stored in data.pos

  • num_pos_edges (int) – Number of positive edges

  • num_neg_edges (int) – Number of negative edges

  • num_edges_perG (int) – Number of edges per graph