sinapsis-data-analysis 0.1.8__py3-none-any.whl → 0.1.9__py3-none-any.whl

sinapsis_data_analysis/templates/ml_base_training.py

@@ -7,6 +7,7 @@ import numpy as np
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base.base_models import TemplateAttributes
 from sinapsis_core.template_base.dynamic_template import BaseDynamicWrapperTemplate
+from sinapsis_core.utils.env_var_keys import WORKING_DIR
 from sinapsis_data_readers.templates.datasets_readers.dataset_splitter import TabularDatasetSplit
 from sklearn.base import is_classifier, is_regressor
 from sklearn.metrics import (
@@ -34,6 +35,7 @@ class MLBaseAttributes(TemplateAttributes):
     """
 
     generic_field_key: str
+    root_dir : str = WORKING_DIR
     model_save_path: str
 
 
@@ -81,7 +83,7 @@ class MLBaseTraining(BaseDynamicWrapperTemplate):
         """
         return dataset is not None
 
-    def process_dataset(self, dataset: TabularDatasetSplit) -> tuple | None:
+    def process_dataset(self, dataset: TabularDatasetSplit | dict) -> tuple | None:
         """
         Extracts x_train, y_train, x_test, y_test from the dataset
 
@@ -92,6 +94,8 @@ class MLBaseTraining(BaseDynamicWrapperTemplate):
             tuple | None: A tuple containing (x_train, y_train, x_test, y_test)
                 or None if the dataset doesn't have the expected attributes
         """
+        if isinstance(dataset, dict):
+            dataset = TabularDatasetSplit(**dataset)
         try:
             x_train = dataset.x_train
             y_train = dataset.y_train
@@ -213,9 +217,9 @@ class MLBaseTraining(BaseDynamicWrapperTemplate):
         if self.trained_model is None:
             self.logger.error("No model to save")
             return
-
+        full_path = os.path.join(self.attributes.root_dir, self.attributes.model_save_path)
         try:
-            os.makedirs(os.path.dirname(self.attributes.model_save_path), exist_ok=True)
+            os.makedirs(os.path.dirname(full_path), exist_ok=True)
             self._save_model_implementation()
             self.logger.info(f"Model saved at {self.attributes.model_save_path}")
         except (MemoryError, TypeError) as e:
@@ -255,7 +259,7 @@ class MLBaseTraining(BaseDynamicWrapperTemplate):
         results = self.handle_model_training(processed_data)
 
         if results is not None:
-            self._set_generic_data(container, results)
+            self._set_generic_data(container, results.model_dump())
             self.save_model()
 
         return container

{sinapsis_data_analysis-0.1.8.dist-info → sinapsis_data_analysis-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-analysis
-Version: 0.1.8
+Version: 0.1.9
 Summary: Templates to work with models for classification, regression and clustering with xgboost and sklearn.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_data_analysis-0.1.8.dist-info → sinapsis_data_analysis-0.1.9.dist-info}/RECORD

@@ -5,14 +5,14 @@ sinapsis_data_analysis/helpers/model_metrics.py,sha256=mR-ZLD8PrGgOh1PrYG8TOjFhJ
 sinapsis_data_analysis/helpers/tags.py,sha256=ICJNmbNAtwn5BfMvGHh6eLuq5Ro3hCIbwd-Y63qOmC0,372
 sinapsis_data_analysis/templates/__init__.py,sha256=z1-E6V4f6MqoGY0_cPwSurDpXDdMOAEOvSS4ogrl1DM,840
 sinapsis_data_analysis/templates/ml_base_inference.py,sha256=CMAebWYKAyNWWdJFEc4zAjFH5z5RlEbsunvYhlunYno,3467
-sinapsis_data_analysis/templates/ml_base_training.py,sha256=sZfYuYo1z19PqoMXKVM1zO_y4BuucuptK6m6eu94o2U,9055
+sinapsis_data_analysis/templates/ml_base_training.py,sha256=3-YfNie6kyWHmpn9Dn2okfpOoGQsjY1iH-a3hwxt5aw,9325
 sinapsis_data_analysis/templates/sklearn_inference.py,sha256=rf-HsWYE4_g9KM_BZbP3f2jKyKz_MLj9-eZ1CAGHqp8,931
 sinapsis_data_analysis/templates/sklearn_manifold.py,sha256=whYTvwDnVacSh4N2XBXVgn8O1dG-KnQsx6htm5DFEbQ,5519
 sinapsis_data_analysis/templates/sklearn_train.py,sha256=WIF7DIslVIRHCmDAXfVUZkiVtLRPRbg6Snq0Di7Ln3w,6757
 sinapsis_data_analysis/templates/xgboost_inference.py,sha256=rFk44jHFCr-EuqDg74TInjlHDCstvA0V422gQmgbd0I,1028
 sinapsis_data_analysis/templates/xgboost_train.py,sha256=VQSztloGAW4Xcx8MZCoRKdqa5jn4NL857jHTeMRuMwA,2513
-sinapsis_data_analysis-0.1.8.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-sinapsis_data_analysis-0.1.8.dist-info/METADATA,sha256=TV0eu_VZ0tIHq79PEhjkw-jz-_LY99SFiK_nDXM6ysA,6296
-sinapsis_data_analysis-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_data_analysis-0.1.8.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
-sinapsis_data_analysis-0.1.8.dist-info/RECORD,,
+sinapsis_data_analysis-0.1.9.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_data_analysis-0.1.9.dist-info/METADATA,sha256=pDujcd32nUV7LV78m8cSCgOm8wDpqo2ztvwkKcI0VZY,6296
+sinapsis_data_analysis-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinapsis_data_analysis-0.1.9.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
+sinapsis_data_analysis-0.1.9.dist-info/RECORD,,