dragon-ml-toolbox 4.4.0__tar.gz → 4.5.0__tar.gz

{dragon_ml_toolbox-4.4.0/dragon_ml_toolbox.egg-info → dragon_ml_toolbox-4.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 4.4.0
+Version: 4.5.0
 Summary: A collection of tools for data science and machine learning projects.
 Author-email: Karl Loza <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-4.4.0 → dragon_ml_toolbox-4.5.0/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 4.4.0
+Version: 4.5.0
 Summary: A collection of tools for data science and machine learning projects.
 Author-email: Karl Loza <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-4.4.0 → dragon_ml_toolbox-4.5.0}/ml_tools/PSO_optimization.py

@@ -2,7 +2,7 @@ import numpy as np
 from pathlib import Path
 import xgboost as xgb
 import lightgbm as lgb
-from typing import Literal, Union, Tuple, Dict, Optional
+from typing import Literal, Union, Tuple, Dict, Optional, Any
 import pandas as pd
 from copy import deepcopy
 from .utilities import (
@@ -25,6 +25,7 @@ from contextlib import nullcontext
 __all__ = [
     "ObjectiveFunction",
     "multiple_objective_functions_from_dir",
+    "parse_lower_upper_bounds",
     "run_pso",
     "plot_optimal_feature_distributions"
 ]
@@ -169,6 +170,18 @@ def multiple_objective_functions_from_dir(directory: Union[str,Path], add_noise:
     return objective_functions, objective_function_names
 
 
+def parse_lower_upper_bounds(source: dict[str,tuple[Any,Any]]):
+    """
+    Parse lower and upper boundaries, returning 2 lists:
+    
+    `lower_bounds`, `upper_bounds`
+    """
+    lower = [low[0] for low in source.values()]
+    upper = [up[1] for up in source.values()]
+    
+    return lower, upper
+
+
 def _set_boundaries(lower_boundaries: list[float], upper_boundaries: list[float]):
     assert len(lower_boundaries) == len(upper_boundaries), "Lower and upper boundaries must have the same length."
     assert len(lower_boundaries) >= 1, "At least one boundary pair is required."

dragon_ml_toolbox-4.5.0/ml_tools/__init__.py

@@ -0,0 +1 @@
+from .custom_logger import custom_logger

{dragon_ml_toolbox-4.4.0 → dragon_ml_toolbox-4.5.0}/ml_tools/data_exploration.py

@@ -348,7 +348,7 @@ def plot_correlation_heatmap(df: pd.DataFrame,
         full_path = save_path / plot_title
         
         plt.savefig(full_path, bbox_inches="tight", format='svg')
-        print(f"Saved correlation heatmap: '{plot_title}.svg'")
+        print(f"Saved correlation heatmap: '{plot_title}'")
     
     plt.show()
     plt.close()

{dragon_ml_toolbox-4.4.0 → dragon_ml_toolbox-4.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dragon-ml-toolbox"
-version = "4.4.0"
+version = "4.5.0"
 description = "A collection of tools for data science and machine learning projects."
 authors = [
     { name = "Karl Loza", email = "luigiloza@gmail.com" }