A Python script that can be used to train and evaluate a GNN for stress prediction.