tumourkit.train_pipe.run_graph_pipe

tumourkit.train_pipe.run_graph_pipe(args: Namespace, logger: Logger) → None

Trains the graph models.

This function trains the graph models using the following steps:

  1. Trains the GCN (Graph Convolutional Network) model.

  2. Trains the GAT (Graph Attention Network) model.

Parameters:

  • args (Namespace) – The arguments for the graph pipeline.

  • logger (Logger) – The logger object used for logging messages.