Module epiclass.utils.shap.shap_to_rank

Script to convert SHAP values files to rank values.

Functions

def main(parent_folder: Path | None = None)

Main

def parse_arguments() ‑> argparse.Namespace

Define CLI argument parser.

def process_split(folder: Path) ‑> Tuple[List[numpy.ndarray], List[str], List[Tuple[str, str]]]

Process a single split folder, extracting SHAP values and associated metadata.

Converts the SHAP values to ranks (for each sample).

Args

folder : Path

Path to the split folder containing SHAP values.

Returns

split_ranks (List[np.ndarray]): List of SHAP rank matrices for each class. eval_md5s (List[str]): List of evaluation md5s. classes (List[Tuple[str, str]]): List of (class_idx, class_name) tuples.