idanmoradarthas / DataScienceUtils / 21175393653

100%

push

github

Add Label Validation to plot_confusion_matrix (#92)

* feat: Add label validation to plot_confusion_matrix

Adds validation to ensure that `y_test`, `y_pred`, and `labels` contain the same set of unique values.

Raises a `ValueError` with a clear error message if there is a mismatch, indicating which values are missing or extra.

Includes a new test function `test_plot_confusion_matrix_label_mismatch` to verify the validation logic.

* refactor: Address feedback from code review

Removes unnecessary `.tolist()` calls in `ds_utils/metrics.py` and fixes the regex pattern in the corresponding test in `tests/test_metrics.py`.

The changes make the code more efficient and the tests more robust.

* docs: Update plot_confusion_matrix docstring

Updates the docstring for the `plot_confusion_matrix` function to describe the new validation behavior and updates the `:raises ValueError:` section accordingly.

---------

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

12 of 12 new or added lines in 1 file covered. (100.0%)

687 of 690 relevant lines covered (99.57%)

11.95 hits per line