dragon-ml-toolbox 14.3.1__py3-none-any.whl → 16.0.0__py3-none-any.whl

{dragon_ml_toolbox-14.3.1.dist-info → dragon_ml_toolbox-16.0.0.dist-info}/METADATA

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 14.3.1
-Summary: A collection of tools for data science and machine learning projects.
-Author-email: "Karl L. Loza Vidaurre" <luigiloza@gmail.com>
+Version: 16.0.0
+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
 Project-URL: Homepage, https://github.com/DrAg0n-BoRn/ML_tools
 Project-URL: Changelog, https://github.com/DrAg0n-BoRn/ML_tools/blob/master/CHANGELOG.md
@@ -79,7 +79,7 @@ Dynamic: license-file
 
 # dragon-ml-toolbox
 
-A collection of Python utilities for data science and machine learning, structured as a modular package for easy reuse and installation. This package has no base dependencies, allowing for lightweight and customized virtual environments.
+A collection of machine learning pipelines and utilities, structured as modular packages for easy reuse and installation. This package has no base dependencies, allowing for lightweight and customized virtual environments.
 
 ### Features:
 
@@ -141,6 +141,7 @@ ETL_cleaning
 ETL_engineering
 math_utilities
 ML_callbacks
+ML_configuration
 ML_datasetmaster
 ML_evaluation_multi
 ML_evaluation
@@ -149,6 +150,10 @@ ML_models
 ML_models_advanced # Requires the extra flag [py-tab]
 ML_optimization
 ML_scaler
+ML_sequence_datasetmaster
+ML_sequence_evaluation
+ML_sequence_inference
+ML_sequence_models
 ML_trainer
 ML_utilities
 ML_vision_datasetmaster
@@ -159,7 +164,6 @@ ML_vision_transformers
 optimization_tools
 path_manager
 PSO_optimization
-RNN_forecast
 serde
 SQL
 utilities
@@ -245,6 +249,7 @@ custom_logger
 GUI_tools
 ML_models
 ML_inference
+ML_sequence_inference
 ML_scaler
 path_manager
 ```

dragon_ml_toolbox-16.0.0.dist-info/RECORD

@@ -0,0 +1,51 @@
+dragon_ml_toolbox-16.0.0.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
+dragon_ml_toolbox-16.0.0.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=8O6qaCbufL9q0I299E9gNeSgZUoqKuSTIiAdKHUldaA,13177
+ml_tools/ML_datasetmaster.py,sha256=j9ZPYULpeJ-XkngNnNHUju49RJD36zFV_HHs58s0c0U,27000
+ml_tools/ML_evaluation.py,sha256=60Fh4IejdXMC-W6hT0MNR5Y3K7L3xYd2aY535Jsf6V4,29207
+ml_tools/ML_evaluation_multi.py,sha256=QDmifXlSzwINUtVTrUyw6g801oYY2INQgKsMUnj3XRc,20155
+ml_tools/ML_inference.py,sha256=CVvX55yhydaSUbwo7Ej63VFK-aeDZ5ttXU4sXY1bGQ0,27482
+ml_tools/ML_models.py,sha256=OEiuUduu2KqsfXZIfzJHR3uop_Zo6dzdKtvaOeRt1G0,27932
+ml_tools/ML_models_advanced.py,sha256=5Y-Kda3P972F9zyfqCS4ndqOL-XXri010nhNp_bhHvY,12411
+ml_tools/ML_optimization.py,sha256=2EwaKHKoZPnvN02d4q0tLO7aBMXSO8cEuhLl0bx28bg,22692
+ml_tools/ML_scaler.py,sha256=Rp6h6U013UK56XhiV_Rmj1CSMI7OSIJLqC0vn6RkiQY,7527
+ml_tools/ML_sequence_datasetmaster.py,sha256=WLkZ_yBcT5bjnbZ1SaecSXBms9IqC596lOCq14D48bc,15569
+ml_tools/ML_sequence_evaluation.py,sha256=AfPkl-92AZ0HYD1xfTSYJSYClvgLYUEW-7CI9ifKB08,7686
+ml_tools/ML_sequence_inference.py,sha256=k9Q8nSvUGdNrmnS4uXh3DkfrxQAqJI68Zs3oaiz2daY,17876
+ml_tools/ML_sequence_models.py,sha256=PVmk7nK-lIl2asR4r6XgT0TIYSJKY4Um4D65gsyE_Qw,5597
+ml_tools/ML_trainer.py,sha256=pZs5rUXgrYt55KX7FoQn-xO14ByDeOf0uS7h329Wn9g,99302
+ml_tools/ML_utilities.py,sha256=kj4CoI7YyAj9fipzFdl9gSYIMzDAjcPqgupTBkB1BQg,22994
+ml_tools/ML_vision_datasetmaster.py,sha256=JiaWfNm34mMQjVgSFOnG_Z4IIkqvvxWfDPQdwPCoN4k,64757
+ml_tools/ML_vision_evaluation.py,sha256=1ZYbB3fuMD5-pynyJuJcF3M19hacepuqVfBbyFTguL8,11331
+ml_tools/ML_vision_inference.py,sha256=8PRqufE2vz_X9NOQs4psI0M7nd3HaSB33vX4VEPCrvk,19618
+ml_tools/ML_vision_models.py,sha256=NojhEZcQiIZ3iKCo5eFkcxetCEVuBKbujVUNruHes-U,26175
+ml_tools/ML_vision_transformers.py,sha256=CEHPzkonub4-s21hjhj30O01dr5sVj9EEkrqmnFl03Y,10749
+ml_tools/PSO_optimization.py,sha256=wAi7BaY-_QoRZ8ibHD6xpyhUABofrabHV7oiryBz5D0,22931
+ml_tools/SQL.py,sha256=hBTKC_OotSuWc0DeD8sI-u2GJS7X_4oANjmLcY1YW_w,11210
+ml_tools/VIF_factor.py,sha256=at5IVqPvicja2-DNSTSIIy3SkzDWCmLzo3qTG_qr5n8,10422
+ml_tools/__init__.py,sha256=kJiankjz9_qXu7gU92mYqYg_anLvt-B6RtW0mMH8uGo,76
+ml_tools/_keys.py,sha256=57_8N-aBxon9ITC7xc00mvJL5czYgx6NVhl0KuL2l3E,4719
+ml_tools/_logger.py,sha256=dlp5cGbzooK9YSNSZYB4yjZrOaQUGW8PTrM411AOvL8,4717
+ml_tools/_schema.py,sha256=bE2RhOhXZd2u8MEQLOM--01ILPDxLqQAhZ3hZpFTXAI,3909
+ml_tools/_script_info.py,sha256=21r83LV3RubsNZ_RTEUON6RbDf7Mh4_udweNcvdF_Fk,212
+ml_tools/constants.py,sha256=3br5Rk9cL2IUo638eJuMOGdbGQaWssaUecYEvSeRBLM,3322
+ml_tools/custom_logger.py,sha256=TGc0Ww2Xlqj2XE3q4bP43hV7T3qnb5ci9f0pYHXF5TY,11226
+ml_tools/data_exploration.py,sha256=bwHzFJ-IAo5GN3T53F-1J_pXUg8VHS91sG_90utAsfg,69911
+ml_tools/ensemble_evaluation.py,sha256=-pxhmCMPjaqSjJxXxaD_asKtoamztATjXJL7YKlsvZk,28369
+ml_tools/ensemble_inference.py,sha256=uQPJiBK1GcckmeWgZn7BzaaPKIAQIBglmbUOuK9WknY,8560
+ml_tools/ensemble_learning.py,sha256=Bh5WupUF93yLM5IBaQBsOqTVjWKjyfz7jN9IuRaZQ_o,21965
+ml_tools/handle_excel.py,sha256=pfdAPb9ywegFkM9T54bRssDOsX-K7rSeV0RaMz7lEAo,14006
+ml_tools/math_utilities.py,sha256=xeKq1quR_3DYLgowcp4Uam_4s3JltUyOnqMOGuAiYWU,8802
+ml_tools/optimization_tools.py,sha256=_sCLZy9LRIIqt1zkYyKNsSbDK3JjRIhC-sADq-JtegE,12751
+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.0.0.dist-info/METADATA,sha256=TT3uGtzGsQDxZkVk-2B6ldAL0f1Xl7mzCi7O9L5Ht9k,6591
+dragon_ml_toolbox-16.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dragon_ml_toolbox-16.0.0.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
+dragon_ml_toolbox-16.0.0.dist-info/RECORD,,

ml_tools/ETL_cleaning.py

@@ -14,8 +14,8 @@ __all__ = [
     "save_unique_values",
     "basic_clean",
     "basic_clean_drop",
-    "ColumnCleaner",
-    "DataFrameCleaner"
+    "DragonColumnCleaner",
+    "DragonDataFrameCleaner"
 ]
 
 
@@ -200,11 +200,11 @@ def _cleaner_core(df_in: pl.DataFrame, all_lowercase: bool) -> pl.DataFrame:
         # Create a cleaner for every column in the dataframe
         all_columns = df_in.columns
         column_cleaners = [
-            ColumnCleaner(col, rules=cleaning_rules, case_insensitive=True) for col in all_columns
+            DragonColumnCleaner(col, rules=cleaning_rules, case_insensitive=True) for col in all_columns
         ]
         
         # Instantiate and run the main dataframe cleaner
-        df_cleaner = DataFrameCleaner(cleaners=column_cleaners)
+        df_cleaner = DragonDataFrameCleaner(cleaners=column_cleaners)
         df_cleaned = df_cleaner.clean(df_in, clone_df=False) # Use clone_df=False for efficiency
         
         # apply lowercase to all string columns
@@ -335,12 +335,12 @@ def basic_clean_drop(input_filepath: Union[str,Path], output_filepath: Union[str
 
 
 ########## EXTRACT and CLEAN ##########
-class ColumnCleaner:
+class DragonColumnCleaner:
     """
     A configuration object that defines cleaning rules for a single Polars DataFrame column.
 
     This class holds a dictionary of regex-to-replacement rules, the target column name,
-    and the case-sensitivity setting. It is intended to be used with the DataFrameCleaner.
+    and the case-sensitivity setting. It is intended to be used with the DragonDataFrameCleaner.
     
     Notes:
         - Define rules from most specific to more general to create a fallback system.
@@ -365,8 +365,8 @@ class ColumnCleaner:
         r'ID[- ](\\d+)': r'ID:$1'
     }
 
-    id_cleaner = ColumnCleaner(column_name='user_id', rules=id_rules)
-    # This object would then be passed to a DataFrameCleaner.
+    id_cleaner = DragonColumnCleaner(column_name='user_id', rules=id_rules)
+    # This object would then be passed to a DragonDataFrameCleaner.
     ```
     """
     def __init__(self, column_name: str, rules: Dict[str, str], case_insensitive: bool = True):
@@ -382,34 +382,34 @@ class ColumnCleaner:
         self.case_insensitive = case_insensitive
 
 
-class DataFrameCleaner:
+class DragonDataFrameCleaner:
     """
     Orchestrates cleaning multiple columns in a Polars DataFrame.
 
-    This class takes a list of ColumnCleaner objects and applies their defined
+    This class takes a list of DragonColumnCleaner objects and applies their defined
     rules to the corresponding columns of a DataFrame using high-performance
     Polars expressions.
 
     Args:
-        cleaners (List[ColumnCleaner]):
-            A list of ColumnCleaner configuration objects.
+        cleaners (List[DragonColumnCleaner]):
+            A list of DragonColumnCleaner configuration objects.
 
     Raises:
-        TypeError: If 'cleaners' is not a list or contains non-ColumnCleaner objects.
-        ValueError: If multiple ColumnCleaner objects target the same column.
+        TypeError: If 'cleaners' is not a list or contains non-DragonColumnCleaner objects.
+        ValueError: If multiple DragonColumnCleaner objects target the same column.
     """
-    def __init__(self, cleaners: List[ColumnCleaner]):
+    def __init__(self, cleaners: List[DragonColumnCleaner]):
         if not isinstance(cleaners, list):
-            _LOGGER.error("The 'cleaners' argument must be a list of ColumnCleaner objects.")
+            _LOGGER.error("The 'cleaners' argument must be a list of DragonColumnCleaner objects.")
             raise TypeError()
 
         seen_columns = set()
         for cleaner in cleaners:
-            if not isinstance(cleaner, ColumnCleaner):
-                _LOGGER.error(f"All items in 'cleaners' list must be ColumnCleaner objects, but found an object of type {type(cleaner).__name__}.")
+            if not isinstance(cleaner, DragonColumnCleaner):
+                _LOGGER.error(f"All items in 'cleaners' list must be DragonColumnCleaner objects, but found an object of type {type(cleaner).__name__}.")
                 raise TypeError()
             if cleaner.column_name in seen_columns:
-                _LOGGER.error(f"Duplicate ColumnCleaner found for column '{cleaner.column_name}'. Each column should only have one cleaner.")
+                _LOGGER.error(f"Duplicate DragonColumnCleaner found for column '{cleaner.column_name}'. Each column should only have one cleaner.")
                 raise ValueError()
             seen_columns.add(cleaner.column_name)
 
@@ -475,7 +475,7 @@ class DataFrameCleaner:
         """
         This convenience method encapsulates the entire cleaning process into a
         single call. It loads a DataFrame from a specified file, applies all
-        cleaning rules configured in the `DataFrameCleaner` instance, and saves
+        cleaning rules configured in the `DragonDataFrameCleaner` instance, and saves
         the resulting cleaned DataFrame to a new file.
 
         The method ensures that all data is loaded as string types to prevent

ml_tools/ETL_engineering.py

@@ -8,11 +8,12 @@ from .path_manager import make_fullpath
 from ._script_info import _script_info
 from ._logger import _LOGGER
 from .constants import CHEMICAL_ELEMENT_SYMBOLS
+from ._keys import MagicWords
 
 
 __all__ = [
-    "TransformationRecipe",
-    "DataProcessor",
+    "DragonTransformRecipe",
+    "DragonProcessor",
     "BinaryTransformer",
     "MultiBinaryDummifier",
     "AutoDummifier",
@@ -32,16 +33,13 @@ __all__ = [
 
 ############ TRANSFORM MAIN ####################
 
-# Magic word for rename-only transformation
-_RENAME = "rename"
-
-class TransformationRecipe:
+class DragonTransformRecipe:
     """
     A builder class for creating a data transformation recipe.
 
     This class provides a structured way to define a series of transformation
     steps, with validation performed at the time of addition. It is designed
-    to be passed to a `DataProcessor`.
+    to be passed to a `DragonProcessor`.
     
     Use the method `add()` to add recipes.
     """
@@ -53,7 +51,7 @@ class TransformationRecipe:
         input_col_name: str,
         transform: Union[str, Callable],
         output_col_names: Optional[Union[str, List[str]]] = None
-    ) -> "TransformationRecipe":
+    ) -> "DragonTransformRecipe":
         """
         Adds a new transformation step to the recipe.
 
@@ -77,12 +75,12 @@ class TransformationRecipe:
             _LOGGER.error("'input_col' must be a non-empty string.")
             raise TypeError()
             
-        if transform == _RENAME:
+        if transform == MagicWords.RENAME:
             if not isinstance(output_col_names, str):
                 _LOGGER.error("For a RENAME operation, 'output_col' must be a string.")
                 raise TypeError()
         elif not isinstance(transform, Callable):
-            _LOGGER.error(f"'transform' must be a callable function or the string '{_RENAME}'.")
+            _LOGGER.error(f"'transform' must be a callable function or the string '{MagicWords.RENAME}'.")
             raise TypeError()
         
         # --- Add Step ---
@@ -103,22 +101,22 @@ class TransformationRecipe:
         return len(self._steps)
 
 
-class DataProcessor:
+class DragonProcessor:
     """
-    Transforms a Polars DataFrame based on a provided `TransformationRecipe` object.
+    Transforms a Polars DataFrame based on a provided `DragonTransformRecipe` object.
     
     Use the methods `transform()` or `load_transform_save()`.
     """
-    def __init__(self, recipe: TransformationRecipe):
+    def __init__(self, recipe: DragonTransformRecipe):
         """
-        Initializes the DataProcessor with a transformation recipe.
+        Initializes the DragonProcessor with a transformation recipe.
 
         Args:
-            recipe: An instance of the `TransformationRecipe` class that has
+            recipe: An instance of the `DragonTransformRecipe` class that has
                     been populated with transformation steps.
         """
-        if not isinstance(recipe, TransformationRecipe):
-            _LOGGER.error("The recipe must be an instance of TransformationRecipe.")
+        if not isinstance(recipe, DragonTransformRecipe):
+            _LOGGER.error("The recipe must be an instance of DragonTransformRecipe.")
             raise TypeError()
         if len(recipe) == 0:
             _LOGGER.error("The recipe cannot be empty.")
@@ -142,7 +140,7 @@ class DataProcessor:
 
             input_series = df.get_column(input_col_name)
 
-            if transform_action == _RENAME:
+            if transform_action == MagicWords.RENAME:
                 processed_columns.append(input_series.alias(output_col_spec))
                 continue
 
@@ -237,7 +235,7 @@ class DataProcessor:
         Provides a detailed, human-readable string representation of the
         entire processing pipeline.
         """
-        header = "DataProcessor Pipeline"
+        header = "DragonProcessor Pipeline"
         divider = "-" * len(header)
         num_steps = len(self._recipe)
         
@@ -255,7 +253,7 @@ class DataProcessor:
             transform_action = step["transform"]
             
             # Get a clean name for the transformation action
-            if transform_action == _RENAME: # "rename"
+            if transform_action == MagicWords.RENAME: # "rename"
                 transform_name = "Rename"
             else:
                 # This works for both functions and class instances
@@ -394,7 +392,7 @@ class MultiBinaryDummifier:
 
     For each keyword provided, this transformer generates a corresponding column
     with a value of 1 if the keyword is present in the input string, and 0 otherwise.
-    It is designed to be used within the DataProcessor pipeline.
+    It is designed to be used within the DragonProcessor pipeline.
 
     Args:
         keywords (List[str]):
@@ -443,7 +441,7 @@ class MultiBinaryDummifier:
                 .when(str_column.str.contains(pattern))
                 .then(pl.lit(1, dtype=pl.UInt8))
                 .otherwise(pl.lit(0, dtype=pl.UInt8))
-                .alias(f"{column_base_name}_{keyword}") # name for DataProcessor
+                .alias(f"{column_base_name}_{keyword}") # name for DragonProcessor
             )
             output_expressions.append(expr)
 
@@ -533,7 +531,7 @@ class NumberExtractor:
     A configurable transformer that extracts a single number from a Polars string series using a regular expression.
 
     An instance can be used as a 'transform' callable within the
-    `DataProcessor` pipeline.
+    `DragonProcessor` pipeline.
 
     Args:
         regex_pattern (str):
@@ -872,7 +870,7 @@ class MultiTemperatureExtractor:
                 pl.when(column.is_not_null())
                 .then(final_expr)
                 .otherwise(None)
-                .alias(f"{column_base_name}_{i}") # Temporary name for DataProcessor
+                .alias(f"{column_base_name}_{i}") # Temporary name for DragonProcessor
             )
             
             output_expressions.append(final_expr)
@@ -1300,7 +1298,7 @@ class MolecularFormulaTransformer:
     each chemical element has its own column. The value in each column is the
     stoichiometric quantity of that element.
 
-    It is designed to be used within the DataProcessor pipeline.
+    It is designed to be used within the DragonProcessor pipeline.
     """
 
     def __init__(self):

ml_tools/GUI_tools.py

@@ -8,15 +8,15 @@ import numpy as np
 
 from ._script_info import _script_info
 from ._logger import _LOGGER
-from .keys import _OneHotOtherPlaceholder
+from ._keys import _OneHotOtherPlaceholder
 
 
 __all__ = [
-    "ConfigManager", 
-    "GUIFactory",
+    "DragonGUIConfig", 
+    "DragonGUIFactory",
     "catch_exceptions", 
-    "FeatureMaster",
-    "GUIHandler" 
+    "DragonFeatureMaster",
+    "DragonGUIHandler" 
 ]
 
 # --- Configuration Management ---
@@ -55,14 +55,14 @@ class _SectionProxy:
         # Fallback to the original string
         return value_str
 
-class ConfigManager:
+class DragonGUIConfig:
     """
     Loads a .ini file and provides access to its values as object attributes.
     Includes a method to generate a default configuration template.
     """
     def __init__(self, config_path: str | Path):
         """
-        Initializes the ConfigManager and dynamically creates attributes
+        Initializes the DragonGUIConfig and dynamically creates attributes
         based on the .ini file's sections and options.
         """
         config_path = Path(config_path)
@@ -78,7 +78,7 @@ class ConfigManager:
     @staticmethod
     def generate_template(file_path: str | Path):
         """
-        Generates a complete, commented .ini template file that works with the GUIFactory.
+        Generates a complete, commented .ini template file that works with the DragonGUIFactory.
 
         Args:
             file_path (str | Path): The path where the .ini file will be saved.
@@ -155,12 +155,12 @@ class ConfigManager:
 
 
 # --- GUI Factory ---
-class GUIFactory:
+class DragonGUIFactory:
     """
     Builds styled FreeSimpleGUI elements and layouts using a "building block"
-    approach, driven by a ConfigManager instance.
+    approach, driven by a DragonGUIConfig instance.
     """
-    def __init__(self, config: ConfigManager):
+    def __init__(self, config: DragonGUIConfig):
         """
         Initializes the factory with a configuration object.
         """
@@ -456,7 +456,7 @@ def catch_exceptions(show_popup: bool = True):
 
 
 # --- Feature Handler ---
-class FeatureMaster:
+class DragonFeatureMaster:
     """
     Manages and organizes feature definitions for a machine learning model.
 
@@ -488,7 +488,7 @@ class FeatureMaster:
                  categorical_features: Optional[List[Tuple[str, str, Dict[str, int]]]] = None,
                  add_one_hot_other_placeholder: bool = True) -> None:
         """
-        Initializes the FeatureMaster instance by processing feature and target definitions.
+        Initializes the DragonFeatureMaster instance by processing feature and target definitions.
 
         This constructor creates internal mappings to translate between GUI-friendly names and model-specific feature names. It also
         prepares data structures needed to populate UI components.
@@ -806,17 +806,17 @@ class FeatureMaster:
 
 
 # --- GUI-Model API ---
-class GUIHandler:
+class DragonGUIHandler:
     """
     Translates data between a GUI and a machine learning model.
     
     This class acts as the primary interface between a user-facing application
-    (FreeSimpleGUI) and the model's expected data format. It uses a `FeatureMaster` instance to correctly process
+    (FreeSimpleGUI) and the model's expected data format. It uses a `DragonFeatureMaster` instance to correctly process
     and encode user inputs.
 
     Its main responsibilities are:
     1.  To take raw values from GUI elements and, using the definitions from
-        `FeatureMaster`, convert them into a single, ordered `numpy.ndarray`
+        `DragonFeatureMaster`, convert them into a single, ordered `numpy.ndarray`
         that can be fed directly into a model for inference.
     2.  To take the results of a model's inference and update the
         corresponding target fields in the GUI to display the prediction.
@@ -824,13 +824,13 @@ class GUIHandler:
     This handler ensures a clean separation of concerns, where the GUI is
     only responsible for presentation, and the model sees correctly formatted numerical data.
     """
-    def __init__(self, feature_handler: FeatureMaster, model_expected_features: list[str]) -> None:
+    def __init__(self, feature_handler: DragonFeatureMaster, model_expected_features: list[str]) -> None:
         """
-        Initializes the GUIHandler.
+        Initializes the DragonGUIHandler.
 
         Args:
-            feature_handler (FeatureMaster):
-                An initialized instance of the `FeatureMaster` class. This object
+            feature_handler (DragonFeatureMaster):
+                An initialized instance of the `DragonFeatureMaster` class. This object
                 contains all the necessary mappings and definitions for the model's
                 features and targets.
             model_expected_features (list[str]):

ml_tools/MICE_imputation.py

@@ -15,7 +15,7 @@ from ._schema import FeatureSchema
 
 
 __all__ = [
-    "MiceImputer",
+    "DragonMICE",
     "apply_mice",
     "save_imputed_datasets",
     "get_convergence_diagnostic",
@@ -280,7 +280,7 @@ def _skip_targets(df: pd.DataFrame, target_cols: list[str]):
 
 
 # modern implementation
-class MiceImputer:
+class DragonMICE:
     """
     A modern MICE imputation pipeline that uses a FeatureSchema
     to correctly discretize categorical features after imputation.
@@ -311,7 +311,7 @@ class MiceImputer:
         # 3. Names of categorical features
         self.categorical_features = list(self.schema.categorical_feature_names)
 
-        _LOGGER.info(f"MiceImputer initialized. Found {len(self.cat_info)} categorical features to discretize.")
+        _LOGGER.info(f"DragonMICE initialized. Found {len(self.cat_info)} categorical features to discretize.")
         
     def _post_process(self, imputed_df: pd.DataFrame) -> pd.DataFrame:
         """