Feature/stack showcase (#83)

* Added mlflow * More refinement of the Jupyter Notebook * Removed airflow * Applied some changes * Further streamlining
zenml-io · Dec 12, 2023 · 88a7653 · 88a7653
1 parent 18ffc39
commit 88a7653
Show file tree

Hide file tree

Showing 5 changed files with 236 additions and 191 deletions.
diff --git a/stack-showcase/pipelines/__init__.py b/stack-showcase/pipelines/__init__.py
@@ -2,4 +2,4 @@
 
 from .feature_engineering import feature_engineering
 from .inference import inference
-from .training import training
+from .training import breast_cancer_training
diff --git a/stack-showcase/pipelines/training.py b/stack-showcase/pipelines/training.py
@@ -14,8 +14,8 @@
 logger = get_logger(__name__)
 
 
-@pipeline
-def training(
+@pipeline(enable_cache=True)
+def breast_cancer_training(
     train_dataset_id: Optional[UUID] = None,
     test_dataset_id: Optional[UUID] = None,
     min_train_accuracy: float = 0.0,