Approach to the Problem

Our approach

We use the PyTorch Geometric library to train the GNN problem, which implements several algorithms.

The first step in this approach is to build a data handler. There are a few methods provided in torch_geometric.data for this purpose.

Custom data handler

Let's take a look at the following code in which we create a custom data handler for our graph data using the PyTorch Geometric library: