sinapsis-data-tools 0.2.25__tar.gz → 0.2.26__tar.gz

{sinapsis_data_tools-0.2.25 → sinapsis_data_tools-0.2.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-tools
-Version: 0.2.25
+Version: 0.2.26
 Summary: Module that contains different packages to perform data-related operations through Sinapsis templates.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_data_tools-0.2.25 → sinapsis_data_tools-0.2.26}/packages/sinapsis_data_analysis/src/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_tools-0.2.25 → sinapsis_data_tools-0.2.26}/packages/sinapsis_data_analysis/src/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
@@ -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_tools-0.2.25 → sinapsis_data_tools-0.2.26}/packages/sinapsis_data_tools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-tools
-Version: 0.2.25
+Version: 0.2.26
 Summary: Module that contains different packages to perform data-related operations through Sinapsis templates.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_data_tools-0.2.25 → sinapsis_data_tools-0.2.26}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sinapsis-data-tools"
-version = "0.2.25"
+version = "0.2.26"
 description = "Module that contains different packages to perform data-related operations through Sinapsis templates."
 authors = [{ name = "SinapsisAI", email = "dev@sinapsis.tech" }]
 dependencies = [