dragon-ml-toolbox 20.1.0__py3-none-any.whl → 20.1.1__py3-none-any.whl

{dragon_ml_toolbox-20.1.0.dist-info → dragon_ml_toolbox-20.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 20.1.0
+Version: 20.1.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-20.1.0.dist-info → dragon_ml_toolbox-20.1.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-dragon_ml_toolbox-20.1.0.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
-dragon_ml_toolbox-20.1.0.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=0-HBRMMgKuwtGy6nMJZvIn1fLxhx_ksyyVB2U_iyYZU,2818
+dragon_ml_toolbox-20.1.1.dist-info/licenses/LICENSE,sha256=L35WDmmLZNTlJvxF6Vy7Uy4SYNi6rCfWUqlTHpoRMoU,1081
+dragon_ml_toolbox-20.1.1.dist-info/licenses/LICENSE-THIRD-PARTY.md,sha256=0-HBRMMgKuwtGy6nMJZvIn1fLxhx_ksyyVB2U_iyYZU,2818
 ml_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ml_tools/constants.py,sha256=3br5Rk9cL2IUo638eJuMOGdbGQaWssaUecYEvSeRBLM,3322
 ml_tools/ETL_cleaning/__init__.py,sha256=TytE8RKmtW4KQlkaTxpYKlJAbCu-VAc82eDdHwVD3Jo,427
@@ -40,7 +40,7 @@ ml_tools/ML_configuration/_finalize.py,sha256=l_n13bLu0avMdJ8hNRrH8V_wOBQZM1UGsT
 ml_tools/ML_configuration/_imprimir.py,sha256=H3h0xYUA6y8LYtO57DIHD0GTX-9b2CaMI2T_3jp5qOo,1452
 ml_tools/ML_configuration/_metrics.py,sha256=PqBGPO1Y_6ImmYI3TEBJhzipULE854vbvE0AbP5m8zQ,22888
 ml_tools/ML_configuration/_models.py,sha256=lvuuqvD6DWUzOa3i06NZfrdfOi9bu2e26T_QO6BGMSw,7629
-ml_tools/ML_configuration/_training.py,sha256=VJcwjkTV4Bvj21jQMnBKgfgaNNXK5yVH5TyFYAF7-N4,5307
+ml_tools/ML_configuration/_training.py,sha256=W-x2J0tqBIVqEkmjfvu7frIuK98JB6aUfG8IZBzdECI,5949
 ml_tools/ML_datasetmaster/__init__.py,sha256=30QTwpZBwZRg5M6CNSjCLNc0vnKFLw0LA7u9Wg0DGOg,517
 ml_tools/ML_datasetmaster/_base_datasetmaster.py,sha256=bwP8_dcpQ-6LvW6ijU5jJCubhI2GrdxavietDC7HmR0,13948
 ml_tools/ML_datasetmaster/_datasetmaster.py,sha256=9B8NdasdeFNMvEMCt5ceq2xRJrB7g6JkgD3KDYFYAJE,18383
@@ -147,7 +147,7 @@ ml_tools/excel_handler/_excel_handler.py,sha256=TODudmeQgDSdxUKzLfAzizs--VL-g8Wx
 ml_tools/excel_handler/_imprimir.py,sha256=QHazgqjRMzthRbDt33EVpvR7GqufSzng6jHw7IVCdtI,306
 ml_tools/keys/__init__.py,sha256=DV52KLOY5GfpLwJdDAHlFVz0qAmyh-KWg3gZorFdMSk,336
 ml_tools/keys/_imprimir.py,sha256=4qmwdia16DPq3OtlWGMkgLPT5R3lcM-ka3tQdCLx5qk,197
-ml_tools/keys/_keys.py,sha256=fArSyT_UGGSH4PHjG-R0kefFznAtAxSAasDCQ7-89a8,8899
+ml_tools/keys/_keys.py,sha256=DLP0BYibRueM_8Dz9pSbWUpKypcimFL5kmXUl4wSwdU,9292
 ml_tools/math_utilities/__init__.py,sha256=NuTcb_Ogdwx5x-oDieBt1EAqCoZRnXbkZbUrwB6ItH0,337
 ml_tools/math_utilities/_imprimir.py,sha256=kk5DQb_BV9g767uTdXQiRjEEHgQwJpEXU3jxO3QV2Fw,238
 ml_tools/math_utilities/_math_utilities.py,sha256=BYHIVcM9tuKIhVrkgLLiM5QalJ39zx7dXYy_M9aGgiM,9012
@@ -173,7 +173,7 @@ ml_tools/utilities/__init__.py,sha256=pkR2HxUIlKZMDderP2awYXVIFxkU2Xt3FkJmcmuRIp
 ml_tools/utilities/_imprimir.py,sha256=sV3ASBOsTdVYvGojOTIpZYFyrnd4panS5h_4HcMzob4,432
 ml_tools/utilities/_utility_save_load.py,sha256=7skiiuYGVLVMK_nU9uLfUZw16ePvF3i9ub7G7LMyUgs,16085
 ml_tools/utilities/_utility_tools.py,sha256=bN0J9d1S0W5wNzNntBWqDsJcEAK7-1OgQg3X2fwXns0,6918
-dragon_ml_toolbox-20.1.0.dist-info/METADATA,sha256=g8BdKr-giBfa-J0TWjinoX1W4lzGaTFZEovm_Fv_43w,7866
-dragon_ml_toolbox-20.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dragon_ml_toolbox-20.1.0.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
-dragon_ml_toolbox-20.1.0.dist-info/RECORD,,
+dragon_ml_toolbox-20.1.1.dist-info/METADATA,sha256=075Nmdtqm7RtKasIDpEoL0U5NFx9pdrSfjg_K10Xzf4,7866
+dragon_ml_toolbox-20.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dragon_ml_toolbox-20.1.1.dist-info/top_level.txt,sha256=wm-oxax3ciyez6VoO4zsFd-gSok2VipYXnbg3TH9PtU,9
+dragon_ml_toolbox-20.1.1.dist-info/RECORD,,

ml_tools/ML_configuration/_training.py

@@ -3,6 +3,7 @@ from pathlib import Path
 
 from .._core import get_logger
 from ..path_manager import make_fullpath
+from ..keys._keys import MLTaskKeys
 
 from ._base_model_config import _BaseModelParams
 
@@ -30,6 +31,9 @@ class DragonTrainingConfig(_BaseModelParams):
                  test_size: float,
                  initial_learning_rate: float,
                  batch_size: int,
+                 task: str,
+                 device: str,
+                 finalized_filename: str,
                  random_state: int = 101,
                  **kwargs: Any) -> None:
         """  
@@ -38,6 +42,9 @@ class DragonTrainingConfig(_BaseModelParams):
             test_size (float): Proportion of data for test set.
             initial_learning_rate (float): Starting learning rate.
             batch_size (int): Number of samples per training batch.
+            task (str): Type of ML task (use TaskKeys).
+            device (str): Device to run training on.
+            finalized_filename (str): Filename for the Dragon ML Finalized-file.
             random_state (int): Seed for reproducibility.
             **kwargs: Additional training parameters as key-value pairs.
         """
@@ -45,7 +52,15 @@ class DragonTrainingConfig(_BaseModelParams):
         self.test_size = test_size
         self.initial_learning_rate = initial_learning_rate
         self.batch_size = batch_size
-        self.random_state = random_state
+        self.device = device
+        self.finalized_filename = finalized_filename
+        self.random_state = random_state        
+        
+        # validate task
+        if task not in MLTaskKeys.ALL_TASKS:
+            _LOGGER.error(f"Invalid task '{task}'. Must be one of: {MLTaskKeys.ALL_TASKS}")
+            raise ValueError()
+        self.task = task
         
         # Process kwargs with validation
         for key, value in kwargs.items():

ml_tools/keys/_keys.py

@@ -198,6 +198,13 @@ class MLTaskKeys:
     SEQUENCE_VALUE = "sequence-to-value"
     
     ALL_BINARY_TASKS = [BINARY_CLASSIFICATION, MULTILABEL_BINARY_CLASSIFICATION, BINARY_IMAGE_CLASSIFICATION, BINARY_SEGMENTATION]
+    
+    ALL_TASKS = [REGRESSION, MULTITARGET_REGRESSION, 
+                 BINARY_CLASSIFICATION, MULTICLASS_CLASSIFICATION, MULTILABEL_BINARY_CLASSIFICATION, 
+                 BINARY_IMAGE_CLASSIFICATION, MULTICLASS_IMAGE_CLASSIFICATION, 
+                 BINARY_SEGMENTATION, MULTICLASS_SEGMENTATION, 
+                 OBJECT_DETECTION, 
+                 SEQUENCE_SEQUENCE, SEQUENCE_VALUE]
 
 
 class _PublicTaskKeys: