sinapsis-data-analysis 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

sinapsis_data_analysis/helpers/excluded_models.py

@@ -24,3 +24,4 @@ excluded_neighbors_models = [
     "LocalOutlierFactor",
 ]
 excluded_tree_models = ["plot_tree", "export_text", "export_graphviz", "BaseDecisionTree"]
+excluded_svm_models = ["l1_min_c"]

sinapsis_data_analysis/templates/ml_base_inference.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+import os
 from abc import abstractmethod
 from typing import Any
 
@@ -6,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.template import Template
+from sinapsis_core.utils.env_var_keys import SINAPSIS_CACHE_DIR
 
 
 class MLBaseInference(Template):
@@ -24,11 +26,12 @@ class MLBaseInference(Template):
         """
 
         model_path: str
+        root_dir : str = SINAPSIS_CACHE_DIR
         generic_field_key: str
 
     def __init__(self, attributes: TemplateAttributes) -> None:
         super().__init__(attributes)
-        self.model = self.load_model(self.attributes.model_path)
+
 
     def get_data(self, container: DataContainer) -> Any:
         """Get the data from the data container
@@ -111,7 +114,8 @@ class MLBaseInference(Template):
         if not self.data_is_valid(data):
             self.logger.warning("Invalid or missing data")
             return container
-
+        full_path = os.path.join(self.attributes.root_dir, self.attributes.model_path)
+        self.model = self.load_model(full_path)
         data = self.preprocess_data(data)
         predictions = self.predict(data)
 

sinapsis_data_analysis/templates/ml_base_training.py

@@ -34,7 +34,7 @@ class MLBaseAttributes(TemplateAttributes):
         model_save_path (str): Path where the trained model will be saved.
     """
 
-    generic_field_key: str
+    generic_field_key: str | None = None
     root_dir : str = WORKING_DIR
     model_save_path: str
 
@@ -220,13 +220,13 @@ class MLBaseTraining(BaseDynamicWrapperTemplate):
         full_path = os.path.join(self.attributes.root_dir, self.attributes.model_save_path)
         try:
             os.makedirs(os.path.dirname(full_path), exist_ok=True)
-            self._save_model_implementation()
+            self._save_model_implementation(full_path)
             self.logger.info(f"Model saved at {self.attributes.model_save_path}")
         except (MemoryError, TypeError) as e:
             self.logger.error(f"Error saving model: {e}")
 
     @abstractmethod
-    def _save_model_implementation(self) -> None:
+    def _save_model_implementation(self, full_path: str) -> None:
         """Save the trained model using an implementation-specific method.
 
         This abstract method should be implemented by subclasses to define

sinapsis_data_analysis/templates/sklearn_train.py

@@ -5,11 +5,12 @@ from sinapsis_core.template_base.base_models import UIPropertiesMetadata
 from sinapsis_core.template_base.dynamic_template import WrapperEntryConfig
 from sinapsis_core.template_base.dynamic_template_factory import make_dynamic_template
 from sinapsis_core.utils.env_var_keys import SINAPSIS_BUILD_DOCS
-from sklearn import linear_model, neighbors, neural_network, tree
+from sklearn import linear_model, neighbors, neural_network, svm, tree
 
 from sinapsis_data_analysis.helpers.excluded_models import (
     excluded_linear_models,
     excluded_neighbors_models,
+    excluded_svm_models,
     excluded_tree_models,
 )
 from sinapsis_data_analysis.helpers.tags import Tags
@@ -53,12 +54,12 @@ class SKLearnLinearModelsTrain(MLBaseTraining):
         category="SKLearn", tags=[Tags.DATA_ANALYSIS, Tags.LINEAR_REGRESSION, Tags.MODELS, Tags.SKLEARN, Tags.TRAINING]
     )
 
-    def _save_model_implementation(self) -> None:
+    def _save_model_implementation(self, full_path:str) -> None:
         """
         Implements the abstract method from the base class to
         save the model to the path specified in attributes.
         """
-        joblib.dump(self.trained_model, self.attributes.model_save_path)
+        joblib.dump(self.trained_model, full_path)
 
 
 class SKLearnNeighborsModelsTrain(SKLearnLinearModelsTrain):
@@ -177,6 +178,41 @@ class SKLearnTreeModelsTrain(SKLearnLinearModelsTrain):
     )
 
 
+class SKLearnSVMModelsTrain(SKLearnLinearModelsTrain):
+    """
+    This template dynamically wraps sklearn's svm module,
+    providing access to models like SVC, SVR, LinearSVC,
+    LinearSVR, NuSVC, NuSVR, and OneClassSVM.
+
+    Usage example:
+
+    agent:
+      name: my_test_agent
+    templates:
+    - template_name: InputTemplate
+      class_name: InputTemplate
+      attributes: {}
+    - template_name: SVCWrapper
+      class_name: SVCWrapper
+      template_input: DataLoaderTemplate
+      attributes:
+        generic_field_key: 'data_loader_key'
+        model_save_path: 'artifacts/svc_model.joblib'
+        svc_init:
+          C: 1.0
+          kernel: rbf
+          random_state: 42
+
+    """
+
+    WrapperEntry = WrapperEntryConfig(
+        wrapped_object=svm,
+        signature_from_doc_string=True,
+        exclude_module_atts=excluded_svm_models,
+        force_init_as_method=False,
+    )
+
+
 def __getattr__(name: str) -> Template:
     """
     Only create a template if it's imported, this avoids creating all the base models for all templates
@@ -190,6 +226,8 @@ def __getattr__(name: str) -> Template:
         return make_dynamic_template(name, SKLearnNNModelsTrain)
     if name in SKLearnTreeModelsTrain.WrapperEntry.module_att_names:
         return make_dynamic_template(name, SKLearnTreeModelsTrain)
+    if name in SKLearnSVMModelsTrain.WrapperEntry.module_att_names:
+        return make_dynamic_template(name, SKLearnSVMModelsTrain)
     raise AttributeError(f"template `{name}` not found in {__name__}")
 
 
@@ -198,6 +236,7 @@ __all__ = (
     + SKLearnNeighborsModelsTrain.WrapperEntry.module_att_names
     + SKLearnNNModelsTrain.WrapperEntry.module_att_names
     + SKLearnTreeModelsTrain.WrapperEntry.module_att_names
+    + SKLearnSVMModelsTrain.WrapperEntry.module_att_names
 )
 
 

{sinapsis_data_analysis-0.1.10.dist-info → sinapsis_data_analysis-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-analysis
-Version: 0.1.10
+Version: 0.1.12
 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
@@ -68,6 +68,7 @@ The following model types are supported:
 - **Linear Models**: LinearRegression, Ridge, Lasso, ElasticNet, LogisticRegression, etc.
 - **Neighbors Models**: KNeighborsClassifier, KNeighborsRegressor, RadiusNeighborsClassifier, etc.
 - **Neural Network Models**: MLPClassifier, MLPRegressor, BernoulliRBM
+- **SVM Models**: SVC, SVR, LinearSVC, LinearSVR, NuSVC, NuSVR, OneClassSVM, etc.
 - **Tree Models**: DecisionTreeClassifier, DecisionTreeRegressor, ExtraTreeClassifier, etc.
 
 Each template uses the same base attributes:

{sinapsis_data_analysis-0.1.10.dist-info → sinapsis_data_analysis-0.1.12.dist-info}/RECORD

@@ -1,18 +1,18 @@
 sinapsis_data_analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_data_analysis/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sinapsis_data_analysis/helpers/excluded_models.py,sha256=gVrTRBdktM3eoBxd7kX22xHlQQS3jPklCdxhisPwsc8,651
+sinapsis_data_analysis/helpers/excluded_models.py,sha256=_95yGsYEiDYXfV5gBlJ1L2XGR1_MebvBS9mGYNy8X2A,686
 sinapsis_data_analysis/helpers/model_metrics.py,sha256=mR-ZLD8PrGgOh1PrYG8TOjFhJz6l2EDRtn6jtMMyI7A,729
 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=3-YfNie6kyWHmpn9Dn2okfpOoGQsjY1iH-a3hwxt5aw,9325
+sinapsis_data_analysis/templates/ml_base_inference.py,sha256=3QfMH8w4yNCnEFZeJOOO3HZAC75UjJL2P6R4aN06f9g,3655
+sinapsis_data_analysis/templates/ml_base_training.py,sha256=m2yDuR9C7lfN1jT31fTQsn3c4fN0Gtx5jd8MtaemUqw,9364
 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/sklearn_train.py,sha256=4b4w2IxoF4ZdYqB1zM9PmMqb7TN8hVvoOtpkPoQTiUc,7870
 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.10.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-sinapsis_data_analysis-0.1.10.dist-info/METADATA,sha256=4LiJ3H5ka3Ocimbxt4F52ohKG2yjgbMVoyOoSzSlcbM,6297
-sinapsis_data_analysis-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_data_analysis-0.1.10.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
-sinapsis_data_analysis-0.1.10.dist-info/RECORD,,
+sinapsis_data_analysis-0.1.12.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_data_analysis-0.1.12.dist-info/METADATA,sha256=k0XwPqShVMV8NhFa4fAMK4-gObfpENHXzd_i2xHonIY,6379
+sinapsis_data_analysis-0.1.12.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+sinapsis_data_analysis-0.1.12.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
+sinapsis_data_analysis-0.1.12.dist-info/RECORD,,

{sinapsis_data_analysis-0.1.10.dist-info → sinapsis_data_analysis-0.1.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any