train_gnn function

Train and Evaluate the Gated Neural Network (robust splits + safe ROC)