tumourkit.classification.infer

Script to generate predictions from given GNN model.

Copyright (C) 2023 Jose Pérez Cano

This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License along with this program. If not, see <https://www.gnu.org/licenses/>.

Contact information: joseperez2000@hotmail.es

Functions

load_normalizer

Loads the normalizers used in training from the given norm_path.

load_saved_model

Loads a saved model from the given weights_path and conf_path.

main

main_with_args

run_inference

Runs inference using the specified model on the provided data loader.

save_probs

Saves the probabilities in .nodes.csv files by appending a column to the original .nodes.csv file.