magnet.graph_utils.ENC_multiple

magnet.graph_utils.ENC_multiple(Y: Tensor, graph: Data) → Tensor

Compute the expected normalized cut for more than 2 partitions.

Parameters:

• Y (torch.Tensor)

• graph (Data) – Graph on which the GNN was evaluated.

Returns:

The value of the loss function.

Return type:

torch.Tensor