dragon-ml-toolbox 19.8.0__tar.gz → 19.8.2__tar.gz

{dragon_ml_toolbox-19.8.0/dragon_ml_toolbox.egg-info → dragon_ml_toolbox-19.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 19.8.0
+Version: 19.8.2
 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
@@ -155,10 +155,11 @@ pip install "dragon-ml-toolbox[ML]"
 
 ```Bash
 constants
-IO_tools
 data_exploration
 ETL_cleaning
 ETL_engineering
+IO_tools
+keys
 math_utilities
 ML_callbacks
 ML_chaining_inference
@@ -313,6 +314,7 @@ pip install "dragon-ml-toolbox[gui-torch]"
 constants
 IO_tools
 GUI_tools
+keys
 ML_models
 ML_models_advanced
 ML_sequence_models

{dragon_ml_toolbox-19.8.0 → dragon_ml_toolbox-19.8.2}/README.md

@@ -55,10 +55,11 @@ pip install "dragon-ml-toolbox[ML]"
 
 ```Bash
 constants
-IO_tools
 data_exploration
 ETL_cleaning
 ETL_engineering
+IO_tools
+keys
 math_utilities
 ML_callbacks
 ML_chaining_inference
@@ -213,6 +214,7 @@ pip install "dragon-ml-toolbox[gui-torch]"
 constants
 IO_tools
 GUI_tools
+keys
 ML_models
 ML_models_advanced
 ML_sequence_models

{dragon_ml_toolbox-19.8.0 → dragon_ml_toolbox-19.8.2/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 19.8.0
+Version: 19.8.2
 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
@@ -155,10 +155,11 @@ pip install "dragon-ml-toolbox[ML]"
 
 ```Bash
 constants
-IO_tools
 data_exploration
 ETL_cleaning
 ETL_engineering
+IO_tools
+keys
 math_utilities
 ML_callbacks
 ML_chaining_inference
@@ -313,6 +314,7 @@ pip install "dragon-ml-toolbox[gui-torch]"
 constants
 IO_tools
 GUI_tools
+keys
 ML_models
 ML_models_advanced
 ML_sequence_models

{dragon_ml_toolbox-19.8.0 → dragon_ml_toolbox-19.8.2}/dragon_ml_toolbox.egg-info/SOURCES.txt

@@ -50,6 +50,7 @@ ml_tools/ensemble_evaluation.py
 ml_tools/ensemble_inference.py
 ml_tools/ensemble_learning.py
 ml_tools/excel_handler.py
+ml_tools/keys.py
 ml_tools/math_utilities.py
 ml_tools/optimization_tools.py
 ml_tools/path_manager.py

{dragon_ml_toolbox-19.8.0 → dragon_ml_toolbox-19.8.2}/ml_tools/_core/_keys.py

@@ -23,6 +23,12 @@ class PyTorchLogKeys:
     BATCH_SIZE = 'size'
 
 
+class _CheckpointCallbackKeys:
+    """Checkpoint callback keys."""
+    TRAIN_LOSS = PyTorchLogKeys.TRAIN_LOSS
+    VALIDATION_LOSS = PyTorchLogKeys.VAL_LOSS
+
+
 class ScalerKeys:
     """Keys for saving/loading scaler artifacts."""
     MEAN = "mean"
@@ -47,7 +53,7 @@ class EnsembleKeys:
 
 
 class PyTorchInferenceKeys:
-    """Keys for the output dictionaries of PyTorchInferenceHandler."""
+    """Keys for the output dictionaries of InferenceHandler classes."""
     # For regression tasks
     PREDICTIONS = "predictions"
     
@@ -120,6 +126,19 @@ class PyTorchCheckpointKeys:
     TARGET_NAMES = "target_names"
 
 
+class _FinalizedFileKeys:
+    """Keys for finalized model files."""
+    MODEL_WEIGHTS = PyTorchCheckpointKeys.MODEL_STATE
+    EPOCH = PyTorchCheckpointKeys.EPOCH
+    TASK = PyTorchCheckpointKeys.TASK
+    CLASSIFICATION_THRESHOLD = PyTorchCheckpointKeys.CLASSIFICATION_THRESHOLD
+    CLASS_MAP = PyTorchCheckpointKeys.CLASS_MAP
+    SEQUENCE_LENGTH = PyTorchCheckpointKeys.SEQUENCE_LENGTH
+    INITIAL_SEQUENCE = PyTorchCheckpointKeys.INITIAL_SEQUENCE
+    TARGET_NAME = PyTorchCheckpointKeys.TARGET_NAME
+    TARGET_NAMES = PyTorchCheckpointKeys.TARGET_NAMES
+
+
 class UtilityKeys:
     """Keys used for utility modules"""
     MODEL_PARAMS_FILE = "model_parameters"

dragon_ml_toolbox-19.8.2/ml_tools/keys.py

@@ -0,0 +1,11 @@
+from ._core._keys import (
+    PyTorchInferenceKeys as InferenceKeys,
+    _CheckpointCallbackKeys as CheckpointCallbackKeys,
+    _FinalizedFileKeys as FinalizedFileKeys,
+)
+
+__all__ = [
+    "InferenceKeys",
+    "CheckpointCallbackKeys",
+    "FinalizedFileKeys",
+]

{dragon_ml_toolbox-19.8.0 → dragon_ml_toolbox-19.8.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dragon-ml-toolbox"
-version = "19.8.0"
+version = "19.8.2"
 description = "Complete pipelines and helper tools for data science and machine learning projects."
 authors = [
     { name = "Karl Luigi Loza Vidaurre", email = "luigiloza@gmail.com" }