Coding the Solution

Solution to the problem

Our aim in this case study is to predict whether a person is infected based on contact tracing information. We use a graph neural network algorithm called ChebNet to build the predictive model.

Let's take a look at the following code, which includes the graph construction, custom data handler, and the training and evaluation of the GNN: