triggerflow 0.1.12__py3-none-any.whl → 0.2.1__py3-none-any.whl

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/model_training/nodes.py

@@ -0,0 +1,31 @@
+"""
+This is a boilerplate pipeline 'model_training'
+generated using Kedro 1.0.0
+"""
+
+import pandas as pd
+from {{ cookiecutter.python_package }}.utils.plotting import plotTrainingHistory, get_dummy
+from {{ cookiecutter.python_package }}.models.{{ cookiecutter.python_package }}_model import {{ cookiecutter.python_package }}
+
+
+def train_model(
+    X_train: pd.DataFrame, y_train: pd.DataFrame, params: dict
+) -> pd.DataFrame:
+    """Trains a dummy model on some data.
+
+    Args:
+        X_train:
+        y_train:
+        params:
+    Returns:
+        Trained model.
+    """
+    params["hps"]["nInputs"] = X_train.shape[-1]
+    model = {{ cookiecutter.python_package }}(name=params["hps"]["name"], hps=params["hps"])
+    model.train(X_train, y_train)
+
+    f, _ = get_dummy()
+    # NOTE: one can also plot the history
+    # f, _ = plotTrainingHistory(model.history)
+
+    return model, f

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/model_training/pipeline.py

@@ -0,0 +1,24 @@
+"""
+This is a boilerplate pipeline 'model_training'
+generated using Kedro 1.0.0
+"""
+
+from kedro.pipeline import node, Pipeline, pipeline  # noqa
+from .nodes import train_model
+
+
+def create_pipeline(**kwargs) -> Pipeline:
+    return pipeline(
+        [
+            node(
+                func=train_model,
+                inputs=[
+                    "processed_{{ cookiecutter.python_package }}_X_train",
+                    "processed_{{ cookiecutter.python_package }}_y_train",
+                    "params:{{ cookiecutter.python_package }}_model",
+                ],
+                outputs=["train_model", "training_history"],
+                name="train_model_node",
+            )
+        ]
+    )

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/model_validation/__init__.py

@@ -0,0 +1,10 @@
+"""
+This is a boilerplate pipeline 'model_validation'
+generated using Kedro 1.0.0
+"""
+
+from .pipeline import create_pipeline
+
+__all__ = ["create_pipeline"]
+
+__version__ = "0.1"

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/model_validation/nodes.py

@@ -0,0 +1,29 @@
+"""
+This is a boilerplate pipeline 'model_validation'
+generated using Kedro 1.0.0
+"""
+
+import logging
+import pandas as pd
+from sklearn.metrics import roc_auc_score
+
+
+def validated_model(model, X_test: pd.DataFrame, y_test: pd.DataFrame) -> pd.DataFrame:
+    """Trains a dummy model on some data.
+
+    Args:
+        model:
+        X_test:
+        y_test:
+    Returns:
+        Model prediction.
+    """
+    # get logger for reporting
+    logger = logging.getLogger(__name__)
+
+    pred = model.predict(X_test)
+    auc = roc_auc_score(y_test, pred)
+
+    logger.info(f"Area under ROC curve: {auc:.4f}")
+
+    return pred

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/pipelines/model_validation/pipeline.py

@@ -0,0 +1,24 @@
+"""
+This is a boilerplate pipeline 'model_validation'
+generated using Kedro 1.0.0
+"""
+
+from kedro.pipeline import node, Pipeline, pipeline  # noqa
+from .nodes import validated_model
+
+
+def create_pipeline(**kwargs) -> Pipeline:
+    return pipeline(
+        [
+            node(
+                func=validated_model,
+                inputs=[
+                    "train_model",
+                    "processed_{{ cookiecutter.python_package }}_X_test",
+                    "processed_{{ cookiecutter.python_package }}_y_test",
+                ],
+                outputs="model_pred",
+                name="validated_model_node",
+            )
+        ]
+    )

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/settings.py

@@ -0,0 +1,46 @@
+"""Project settings. There is no need to edit this file unless you want to change values
+from the Kedro defaults. For further information, including these default values, see
+https://docs.kedro.org/en/stable/kedro_project_setup/settings.html."""
+
+# Instantiated project hooks.
+# For example, after creating a hooks.py and defining a ProjectHooks class there, do
+# from {{ cookiecutter.python_package }}.hooks import ProjectHooks
+# Hooks are executed in a Last-In-First-Out (LIFO) order.
+# HOOKS = (ProjectHooks(),)
+
+# Installed plugins for which to disable hook auto-registration.
+# DISABLE_HOOKS_FOR_PLUGINS = ("kedro-viz",)
+
+# Class that manages storing KedroSession data.
+# from kedro.framework.session.store import BaseSessionStore
+# SESSION_STORE_CLASS = BaseSessionStore
+# Keyword arguments to pass to the `SESSION_STORE_CLASS` constructor.
+# SESSION_STORE_ARGS = {
+#     "path": "./sessions"
+# }
+
+# Directory that holds configuration.
+# CONF_SOURCE = "conf"
+
+# Class that manages how configuration is loaded.
+# from kedro.config import OmegaConfigDataset
+
+# CONFIG_DATASET_CLASS = OmegaConfigDataset
+
+# Keyword arguments to pass to the `CONFIG_DATASET_CLASS` constructor.
+CONFIG_DATASET_ARGS = {
+    "base_env": "base",
+    "default_run_env": "local",
+    # "config_patterns": {
+    #     "spark" : ["spark*/"],
+    #     "parameters": ["parameters*", "parameters*/**", "**/parameters*"],
+    # }
+}
+
+# Class that manages Kedro's library components.
+# from kedro.framework.context import KedroContext
+# CONTEXT_CLASS = KedroContext
+
+# Class that manages the Data Catalog.
+# from kedro.io import DataCatalog
+# DATA_CATALOG_CLASS = DataCatalog

triggerflow/starter/{{ cookiecutter.repo_name }}/src/{{ cookiecutter.python_package }}/utils/metric.py

@@ -0,0 +1,4 @@
+"""
+This is a boilerplate file for custom metrics
+generated using Kedro 1.0.0
+"""