dragon-ml-toolbox 20.1.0__tar.gz → 20.1.1__tar.gz

{dragon_ml_toolbox-20.1.0/dragon_ml_toolbox.egg-info → dragon_ml_toolbox-20.1.1}/PKG-INFO

@@ -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 → dragon_ml_toolbox-20.1.1/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -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 → dragon_ml_toolbox-20.1.1}/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():

{dragon_ml_toolbox-20.1.0 → dragon_ml_toolbox-20.1.1}/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:

{dragon_ml_toolbox-20.1.0 → dragon_ml_toolbox-20.1.1}/pyproject.toml

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