tumourkit.classification.train_graphs.generate_configurations

tumourkit.classification.train_graphs.generate_configurations(max_confs: int, model_name: str) → List[Dict[str, int]]

Generates a grid in the search space with no more than max_confs configurations. Parameters changed: NUM_LAYERS, DROPOUT, NORM_TYPE