dragon-ml-toolbox 16.2.0__py3-none-any.whl → 16.2.1__py3-none-any.whl

{dragon_ml_toolbox-16.2.0.dist-info → dragon_ml_toolbox-16.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 16.2.0
+Version: 16.2.1
 Summary: Complete pipelines and helper tools for data science and machine learning projects.
 Author-email: Karl Luigi Loza Vidaurre <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-16.2.0.dist-info → dragon_ml_toolbox-16.2.1.dist-info}/RECORD

@@ -1,13 +1,13 @@
-dragon_ml_toolbox-16.2.0.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
-dragon_ml_toolbox-16.2.0.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=gkOdNDbKYpIJezwSo2CEnISkLeYfYHv9t8b5K2-P69A,2687
+dragon_ml_toolbox-16.2.1.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
+dragon_ml_toolbox-16.2.1.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=gkOdNDbKYpIJezwSo2CEnISkLeYfYHv9t8b5K2-P69A,2687
 ml_tools/ETL_cleaning.py,sha256=Bg0nTmpNzQKDdezK3m0NjYT7N8_ANGlmD9mDXjggqkA,20522
 ml_tools/ETL_engineering.py,sha256=PGXvlvMWa05J1rsMNXxnHzXIe2K68qhtigSn74W8kFI,54961
 ml_tools/GUI_tools.py,sha256=QMSu-8eSNminD6A6Yg9sXo4ff6GNPThwRBVgQQwAAbY,45508
 ml_tools/MICE_imputation.py,sha256=2MsHeKTd8MSBIYmj0q671Fm4wCBvMGjpxULp__jDNgo,20812
 ml_tools/ML_callbacks.py,sha256=EF7Px_IV3IIJpfaT0Nwbv4-_0C6IUlJ_xjzHOekXwq0,16410
-ml_tools/ML_configuration.py,sha256=MKuy1v53LDLX_a8TDn5DwIdv_tyS2pz44ycw0S6aQfY,31601
+ml_tools/ML_configuration.py,sha256=W4KY4SrpIQAKCmLfVntTWW8fsEuVpHz-CXf_rnNNGqM,31905
 ml_tools/ML_datasetmaster.py,sha256=isvRXI8vNRTFNCFFFpGtsUA8hS6ZDNezLuDpKd9VU9c,28514
-ml_tools/ML_evaluation.py,sha256=LrvTnrS32pFmmsmh_3KGHUREUCNSI5vKIB2JIuBq8oI,30107
+ml_tools/ML_evaluation.py,sha256=eFrOCmETRr1FnbwPk6fbflNXEBLqnnLBWjAI5LmF3dg,30576
 ml_tools/ML_evaluation_multi.py,sha256=mEN8jKaU1N7UdgldEykqME0MV_yubojD1StyQC5bFEA,20416
 ml_tools/ML_inference.py,sha256=qxoeurcqp-soapfgHUuzt-NFg0KGwg_wOIuzsRMyJqQ,29447
 ml_tools/ML_models.py,sha256=OEiuUduu2KqsfXZIfzJHR3uop_Zo6dzdKtvaOeRt1G0,27932
@@ -45,7 +45,7 @@ ml_tools/optimization_tools.py,sha256=_sCLZy9LRIIqt1zkYyKNsSbDK3JjRIhC-sADq-Jteg
 ml_tools/path_manager.py,sha256=2lTnhfDNdYlrqP_LGDoP51LdUf9hlTsZKuZJoYq5W-U,18462
 ml_tools/serde.py,sha256=c8uDYjYry_VrLvoG4ixqDj5pij88lVn6Tu4NHcPkwDU,6943
 ml_tools/utilities.py,sha256=wFwdv7xFV8Sv6kNy4_tE7RNasRs_318Zm7s65Uwu2Us,22509
-dragon_ml_toolbox-16.2.0.dist-info/METADATA,sha256=AX3k2aBOqU4TVzZxagmFQ-NZTJLKa7hpTK7qb0YasuM,6591
-dragon_ml_toolbox-16.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dragon_ml_toolbox-16.2.0.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
-dragon_ml_toolbox-16.2.0.dist-info/RECORD,,
+dragon_ml_toolbox-16.2.1.dist-info/METADATA,sha256=Rr23JuJbUJyYhM-GugIGAjSEde7hj7HL3mCppKU1zCA,6591
+dragon_ml_toolbox-16.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dragon_ml_toolbox-16.2.1.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
+dragon_ml_toolbox-16.2.1.dist-info/RECORD,,

ml_tools/ML_configuration.py

@@ -39,7 +39,7 @@ class _BaseClassificationFormat:
     [PRIVATE] Base configuration for single-label classification metrics.
     """
     def __init__(self, 
-                 cmap: str="Blues",
+                 cmap: str="BuGn",
                  ROC_PR_line: str='darkorange',
                  calibration_bins: int=15, 
                  font_size: int=16) -> None:
@@ -64,7 +64,11 @@ class _BaseClassificationFormat:
         
         <br>
         
-        ## [Matplotlib Colormaps](https://matplotlib.org/stable/users/explain/colors/colormaps.html)
+        ### [Matplotlib Colormaps](https://matplotlib.org/stable/users/explain/colors/colormaps.html)
+        
+        <br>
+        
+        ### [Matplotlib Colors](https://matplotlib.org/stable/gallery/color/named_colors.html)
         """
         self.cmap = cmap
         self.ROC_PR_line = ROC_PR_line
@@ -86,29 +90,33 @@ class _BaseMultiLabelFormat:
     [PRIVATE] Base configuration for multi-label binary classification metrics.
     """
     def __init__(self,
+                 cmap: str = "BuGn",
                  ROC_PR_line: str='darkorange',
-                 cmap: str = "Blues",
                  font_size: int = 16) -> None:
         """
         Initializes the formatting configuration for multi-label classification metrics.
 
         Args:
+            cmap (str): The matplotlib colormap name for the per-label
+                    confusion matrices. Defaults to "Blues".
+                    - Sequential options: 'Blues', 'Greens', 'Reds', 'Oranges', 'Purples'
+                    - Diverging options: 'coolwarm', 'viridis', 'plasma', 'inferno'
+        
             ROC_PR_line (str): The color name or hex code for the line plotted
                 on the ROC and Precision-Recall curves (one for each label). 
                 Defaults to 'darkorange'.
                 - Common color names: 'darkorange', 'cornflowerblue', 'crimson', 'forestgreen'
                 - Hex codes: '#FF6347', '#4682B4'
             
-            cmap (str): The matplotlib colormap name for the per-label
-                confusion matrices. Defaults to "Blues".
-                - Sequential options: 'Blues', 'Greens', 'Reds', 'Oranges', 'Purples'
-                - Diverging options: 'coolwarm', 'viridis', 'plasma', 'inferno'
-            
             font_size (int): The base font size to apply to the plots. Defaults to 16.
             
         <br>
         
-        ## [Matplotlib Colormaps](https://matplotlib.org/stable/users/explain/colors/colormaps.html)    
+        ### [Matplotlib Colormaps](https://matplotlib.org/stable/users/explain/colors/colormaps.html)
+        
+        <br>
+        
+        ### [Matplotlib Colors](https://matplotlib.org/stable/gallery/color/named_colors.html)
         """
         self.cmap = cmap
         self.ROC_PR_line = ROC_PR_line
@@ -116,8 +124,8 @@ class _BaseMultiLabelFormat:
         
     def __repr__(self) -> str:
         parts = [
-            f"ROC_PR_line='{self.ROC_PR_line}'",
             f"cmap='{self.cmap}'",
+            f"ROC_PR_line='{self.ROC_PR_line}'",
             f"font_size={self.font_size}"
         ]
         return f"{self.__class__.__name__}({', '.join(parts)})"
@@ -154,7 +162,7 @@ class _BaseRegressionFormat:
         
         <br>
         
-        ## [Matplotlib Colors](https://matplotlib.org/stable/users/explain/colors/colors.html)
+        ### [Matplotlib Colors](https://matplotlib.org/stable/gallery/color/named_colors.html)
         """
         self.font_size = font_size
         self.scatter_color = scatter_color
@@ -180,8 +188,8 @@ class _BaseSegmentationFormat:
     [PRIVATE] Base configuration for segmentation metrics.
     """
     def __init__(self,
-                 heatmap_cmap: str = 'viridis',
-                 cm_cmap: str = "Blues",
+                 heatmap_cmap: str = "BuGn",
+                 cm_cmap: str = "Purples",
                  font_size: int = 16) -> None:
         """
         Initializes the formatting configuration for segmentation metrics.
@@ -198,7 +206,7 @@ class _BaseSegmentationFormat:
         
         <br>
         
-        ## [Matplotlib Colormaps](https://matplotlib.org/stable/users/explain/colors/colormaps.html)
+        ### [Matplotlib Colormaps](https://matplotlib.org/stable/users/explain/colors/colormaps.html)
         """
         self.heatmap_cmap = heatmap_cmap
         self.cm_cmap = cm_cmap
@@ -241,10 +249,10 @@ class _BaseSequenceValueFormat:
             hist_bins (int | str): The number of bins for the residuals histogram. 
                 Defaults to 'auto' to use seaborn's automatic bin selection.
                 - Options: 'auto', 'sqrt', 10, 20
-        
+
         <br>
         
-        ## [Matplotlib Colors](https://matplotlib.org/stable/users/explain/colors/colors.html)
+        ### [Matplotlib Colors](https://matplotlib.org/stable/gallery/color/named_colors.html)
         """
         self.font_size = font_size
         self.scatter_color = scatter_color
@@ -296,9 +304,15 @@ class _BaseSequenceSequenceFormat:
         
         <br>
         
-        ## [Matplotlib Colors](https://matplotlib.org/stable/users/explain/colors/colors.html)
-        ## [Matplotlib Linestyles](https://matplotlib.org/stable/gallery/lines_bars_and_markers/linestyles.html)
-        ## [Matplotlib Markers](https://matplotlib.org/stable/api/markers_api.html)
+        ### [Matplotlib Colors](https://matplotlib.org/stable/gallery/color/named_colors.html)
+        
+        <br>
+        
+        ### [Matplotlib Linestyles](https://matplotlib.org/stable/gallery/lines_bars_and_markers/linestyles.html)
+        
+        <br>
+        
+        ### [Matplotlib Markers](https://matplotlib.org/stable/api/markers_api.html)
         """
         self.font_size = font_size
         self.plot_figsize = plot_figsize
@@ -366,7 +380,7 @@ class BinaryClassificationMetricsFormat(_BaseClassificationFormat):
     Configuration for binary classification.
     """
     def __init__(self, 
-                 cmap: str="Blues",
+                 cmap: str="BuGn",
                  ROC_PR_line: str='darkorange',
                  calibration_bins: int=15, 
                  font_size: int=16) -> None:
@@ -381,7 +395,7 @@ class MultiClassClassificationMetricsFormat(_BaseClassificationFormat):
     Configuration for multi-class classification.
     """
     def __init__(self, 
-                 cmap: str="Blues",
+                 cmap: str="BuGn",
                  ROC_PR_line: str='darkorange',
                  calibration_bins: int=15, 
                  font_size: int=16) -> None:
@@ -396,7 +410,7 @@ class BinaryImageClassificationMetricsFormat(_BaseClassificationFormat):
     Configuration for binary image classification.
     """
     def __init__(self, 
-                 cmap: str="Blues",
+                 cmap: str="BuGn",
                  ROC_PR_line: str='darkorange',
                  calibration_bins: int=15, 
                  font_size: int=16) -> None:
@@ -411,7 +425,7 @@ class MultiClassImageClassificationMetricsFormat(_BaseClassificationFormat):
     Configuration for multi-class image classification.
     """
     def __init__(self, 
-                 cmap: str="Blues",
+                 cmap: str="BuGn",
                  ROC_PR_line: str='darkorange',
                  calibration_bins: int=15, 
                  font_size: int=16) -> None:
@@ -427,11 +441,11 @@ class MultiLabelBinaryClassificationMetricsFormat(_BaseMultiLabelFormat):
     Configuration for multi-label binary classification.
     """
     def __init__(self,
+                 cmap: str = "BuGn",
                  ROC_PR_line: str='darkorange',
-                 cmap: str = "Blues",
                  font_size: int = 16) -> None:
-        super().__init__(ROC_PR_line=ROC_PR_line, 
-                         cmap=cmap, 
+        super().__init__(cmap=cmap,
+                         ROC_PR_line=ROC_PR_line, 
                          font_size=font_size)
 
 
@@ -441,8 +455,8 @@ class BinarySegmentationMetricsFormat(_BaseSegmentationFormat):
     Configuration for binary segmentation.
     """
     def __init__(self,
-                 heatmap_cmap: str = 'viridis',
-                 cm_cmap: str = "Blues",
+                 heatmap_cmap: str = "BuGn",
+                 cm_cmap: str = "Purples",
                  font_size: int = 16) -> None:
         super().__init__(heatmap_cmap=heatmap_cmap, 
                          cm_cmap=cm_cmap, 
@@ -454,8 +468,8 @@ class MultiClassSegmentationMetricsFormat(_BaseSegmentationFormat):
     Configuration for multi-class segmentation.
     """
     def __init__(self,
-                 heatmap_cmap: str = 'viridis',
-                 cm_cmap: str = "Blues",
+                 heatmap_cmap: str = "BuGn",
+                 cm_cmap: str = "Purples",
                  font_size: int = 16) -> None:
         super().__init__(heatmap_cmap=heatmap_cmap, 
                          cm_cmap=cm_cmap, 

ml_tools/ML_evaluation.py

@@ -169,16 +169,29 @@ def classification_metrics(save_dir: Union[str, Path],
 
     # --- Save Classification Report Heatmap ---
     try:
-        plt.figure(figsize=(8, 6), dpi=DPI_value)
+        # Create DataFrame from report
+        report_df = pd.DataFrame(report_dict)
+        
+        # 1. Drop the 'accuracy' column (single float)
+        if 'accuracy' in report_df.columns:
+            report_df = report_df.drop(columns=['accuracy'])
+        
+        # 2. Select all metric rows *except* the last one ('support')
+        # 3. Transpose the DataFrame
+        plot_df = report_df.iloc[:-1, :].T
+        
+        fig_height = max(5.0, len(plot_df.index) * 0.5 + 2.0)
+        plt.figure(figsize=(7, fig_height), dpi=DPI_value)
+
         sns.set_theme(font_scale=1.2) # Scale seaborn font
-        sns.heatmap(pd.DataFrame(report_dict).iloc[:-1, :].T, 
+        sns.heatmap(plot_df, 
                     annot=True, 
                     cmap=format_config.cmap, 
                     fmt='.2f',
                     vmin=0.0,
                     vmax=1.0)
         sns.set_theme(font_scale=1.0) # Reset seaborn scale
-        plt.title("Classification Report")
+        plt.title("Classification Report Heatmap")
         plt.tight_layout()
         heatmap_path = save_dir_path / "classification_report_heatmap.svg"
         plt.savefig(heatmap_path)