sinapsis-data-analysis 0.1.11__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/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_analysis-0.1.11.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.11
+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.11.dist-info → sinapsis_data_analysis-0.1.12.dist-info}/RECORD

@@ -1,6 +1,6 @@
 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
@@ -8,11 +8,11 @@ sinapsis_data_analysis/templates/ml_base_inference.py,sha256=3QfMH8w4yNCnEFZeJOO
 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=uReIOKQTHYbfsdNy1efVcguwsMTkhDXTbBpSeVDlP-E,6750
+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.11.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-sinapsis_data_analysis-0.1.11.dist-info/METADATA,sha256=OmUDyt8OsaKGfmEmytn_7bgi6pQLpGwVudOrqyauu_I,6297
-sinapsis_data_analysis-0.1.11.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-sinapsis_data_analysis-0.1.11.dist-info/top_level.txt,sha256=Mc5OyqBINgXFLrAyVBmjg25MQd6Lbg7z-rwotzEeygQ,23
-sinapsis_data_analysis-0.1.11.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,,