tumourkit.classification

tumourkit.classification.compute_imbalance

Auxiliar script to compute how imbalanced the dataset is.

tumourkit.classification.evaluate

Script to evaluate predictions.

tumourkit.classification.infer

Script to generate predictions from given GNN model.

tumourkit.classification.models

tumourkit.classification.read_graph

Module to create graph from nodes.

tumourkit.classification.train_graphs

Script to train and save several GNN configurations.

tumourkit.classification.train_xgboost

Script to train xgboost models.