dragon-ml-toolbox 19.8.0__py3-none-any.whl → 19.8.2__py3-none-any.whl

{dragon_ml_toolbox-19.8.0.dist-info → dragon_ml_toolbox-19.8.2.dist-info}/METADATA

@@ -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.dist-info → dragon_ml_toolbox-19.8.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-dragon_ml_toolbox-19.8.0.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
-dragon_ml_toolbox-19.8.0.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=XBLtvGjvBf-q93a5iylHj94Lm78UzInC-3Cii01jc6I,3127
+dragon_ml_toolbox-19.8.2.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
+dragon_ml_toolbox-19.8.2.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=XBLtvGjvBf-q93a5iylHj94Lm78UzInC-3Cii01jc6I,3127
 ml_tools/ETL_cleaning.py,sha256=cKXyRFaaFs_beAGDnQM54xnML671kq-yJEGjHafW-20,351
 ml_tools/ETL_engineering.py,sha256=cwh1FhtNdUHllUDvho-x3SIVj4KwG_rFQR6VYzWUg0U,898
 ml_tools/GUI_tools.py,sha256=O89rG8WQv6GY1DiphQjIsPzXFCQID6te7q_Sgt1iTkQ,294
@@ -43,6 +43,7 @@ ml_tools/ensemble_evaluation.py,sha256=P26vyS2fMV3Pm_4w2MN1z1eS7aVJzYagsyLmqC-Io
 ml_tools/ensemble_inference.py,sha256=sl_Dq9KaN0SrtZmyiVrrhWd6lSjdQangSIUUUIFvfj4,178
 ml_tools/ensemble_learning.py,sha256=BLPnpfJWCly-D75mkRP1FE5TExoWAAlAHR89KAzW9iU,336
 ml_tools/excel_handler.py,sha256=h35HMNnO44btxsTSfZXj2HiJtpRS4fdrJLbzru4heMs,453
+ml_tools/keys.py,sha256=s9HEIAJCRw4DO7ll0yjc8u5rrSI9MOmfkR_1fKpkfy8,263
 ml_tools/math_utilities.py,sha256=53nOXlhb5taUHj4CDHsXliArEfPkOlJD7G_dJa3_iOU,321
 ml_tools/optimization_tools.py,sha256=YQZiXT86kP28NEcyLDbowGeNXHgJDm9hzl-ClNgGQXU,389
 ml_tools/path_manager.py,sha256=ion-x2W_rQjra3ChuOHwVtgXhv7LkpXP0lkBef730tk,350
@@ -92,7 +93,7 @@ ml_tools/_core/_ensemble_evaluation.py,sha256=17lWl4bWLT1BAMv_fhGf2D3wy-F4jx0Hgn
 ml_tools/_core/_ensemble_inference.py,sha256=PfZG-r65Vw3IAmBJZg9W0zYGEe-QbhfUh_rd2ho-rr8,8610
 ml_tools/_core/_ensemble_learning.py,sha256=X8ghbjDOLMENCWdISXLhDlHQtR3C6SW1tkTBAcfRRPY,22016
 ml_tools/_core/_excel_handler.py,sha256=gV4rSIsiowb0xllpEJxzUKaYDDVpmP_lxs9wZA76-cc,14050
-ml_tools/_core/_keys.py,sha256=UpTLHMG1j4FB7hCItcqnfAAuSVMK3Rf-i7jcu6Wkf-Y,6836
+ml_tools/_core/_keys.py,sha256=pOqxhEFcDuAeuQveJNykdQfB6gVEg8ZY7L7MYQmtY_o,7551
 ml_tools/_core/_logger.py,sha256=86Ge0sDE_WgwsZBglQRYPyFYX3lcsIo0NzszNPzlxuk,5254
 ml_tools/_core/_math_utilities.py,sha256=IlXAiZgTcLtus03jJOBOyF9ZCQDf8qLGjrCHu9Mrgak,9091
 ml_tools/_core/_models_advanced_base.py,sha256=ceW0V_CcfOnSFqHlxUhVU8-5mtQq4tFyo8TX-xVexrY,4982
@@ -104,7 +105,7 @@ ml_tools/_core/_schema.py,sha256=TM5WVVMoKOvr_Bc2z34sU_gzKlM465PRKTgdZaEOkGY,140
 ml_tools/_core/_script_info.py,sha256=21r83LV3RubsNZ_RTEUON6RbDf7Mh4_udweNcvdF_Fk,212
 ml_tools/_core/_serde.py,sha256=tsI4EO2Y7jrBMmbQ1pinDsPOrOg-SaPuB-Dt40q0taE,5609
 ml_tools/_core/_utilities.py,sha256=iA8fLWdhsIx4ut2Dp8M_OyU0Y3PPLgGdIklyl17x6xk,22560
-dragon_ml_toolbox-19.8.0.dist-info/METADATA,sha256=ywnJLv63NUVz3LMgHTSbOsRZl95rW02_Sozqwq2u-p0,8764
-dragon_ml_toolbox-19.8.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dragon_ml_toolbox-19.8.0.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
-dragon_ml_toolbox-19.8.0.dist-info/RECORD,,
+dragon_ml_toolbox-19.8.2.dist-info/METADATA,sha256=PBht866esAUBXiPNsjgNFLcwiTlWk0yNFa2RL1QoV8Y,8774
+dragon_ml_toolbox-19.8.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dragon_ml_toolbox-19.8.2.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
+dragon_ml_toolbox-19.8.2.dist-info/RECORD,,

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"

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",
+]