tumourkit.classification.train_graphs.load_model

tumourkit.classification.train_graphs.load_model(conf: Dict[str, Any], num_classes: int, num_feats: int, enable_background: bool) → Module

Available models: GCN, ATT, HATT, SAGE, GIN Configuration space: NUM_LAYERS, DROPOUT, NORM_TYPE