tumourkit.classification.train_graphs.save_model

tumourkit.classification.train_graphs.save_model(save_dir: str, model: Module, conf: Dict[str, Any], normalizers: Tuple[Normalizer], prefix: str | None = '') → None

Save model weights and configuration file to SAVE_DIR