probly.visualization.credal.input_handling

Manages input as well as dispatch to correct plot.

Functions

check_num_classes(input_data)	Checks number of classes.
check_shape(input_data)	Sanity check for input data.
dispatch_plot(input_data[, labels, title, ...])	Selects and executes the correct plotting function based on class count.
normalize_input(input_data)	Normalizes input data.

probly.visualization.credal.input_handling.check_num_classes(input_data)

Checks number of classes.

Parameters:

input_data (ndarray) – Array with last dimension equal to the number of classes.

Returns:

Number of classes.

Return type:

int

probly.visualization.credal.input_handling.check_shape(input_data)

Sanity check for input data.

Parameters:

input_data (ndarray) – Minimum 2D NumPy array with probability vector.

Returns:

input_data or Error Message.

Return type:

ndarray

probly.visualization.credal.input_handling.dispatch_plot(input_data, labels=None, title=None, choice=None, minmax=None)

Selects and executes the correct plotting function based on class count.

Parameters:

• input_data (np.ndarray) – Probabilities vector.

• labels (list[str] | None) – List of labels corresponding to the classes.

• title (str | None) – Manages custom or predefined title.

• choice (str | None) – Allows either “MLE”, “Credal”, “Probability” or None.

• minmax (bool | None) – Enables to show the Min/Max lines for ternary plots.

Return type:

Axes

probly.visualization.credal.input_handling.normalize_input(input_data)

Normalizes input data.

Parameters:

input_data (ndarray) – Array with last dimension equal to the number of classes.

Returns:

2D NumPy array with normalized input data.

Return type:

ndarray