Skip to main content

Module helpers

Function calculate_pirads_performance​

View Source

calculate_pirads_performance(csv: DataFrame) -> dict

Calculates the PIRADS performance.

Args: csv (pd.DataFrame): The csv file containing the PIRADS predictions and groundtruth values.

Returns: dict: The PIRADS performance metrics.

Function save_results_as_csv​

View Source

save_results_as_csv(y_true: UnionType[Series, ndarray], y_proba: UnionType[Series, ndarray], out_path: str, other_features: UnionType[DataFrame, None] = None) -> None

Save the results as a CSV file. Args: y_true (pd.Series | np.ndarray): Ground truth labels y_proba (pd.Series | np.ndarray): Predicted probabilities out_path (str): Path where to save the CSV file other_features (pd.DataFrame | None, optional): Additional features to include. Defaults to None.