refinement of workflow templates

Author: ppalmes

argo-workflow/automlai-dualsearch-template.yaml

@@ -12,18 +12,10 @@ spec:
         value: iris_reg.csv
       - name: predictiontype
         value: regression
-      - name: complexity
-        value: low
-      - name: outputloc
-        value: NONE
       - name: folds
         value: 3
       - name: workers
         value: 5
-      - name: predict_only
-        value: false
-      - name: runid
-        value: NONE
   templates:
     - name: run-main
       inputs:
@@ -83,7 +75,7 @@ spec:
             s3:
               key: csv/
       container:
-        image: ppalmes/automlai:v2.0
+        image: ppalmes/automlai:v3.0
         command:
         args:
           - "-t{{inputs.parameters.predictiontype}}"

argo-workflow/automlai-prediction-classification-template.yaml

@@ -0,0 +1,38 @@
+apiVersion: argoproj.io/v1alpha1
+kind: ClusterWorkflowTemplate
+metadata:
+  name: automlai-prediction-classification
+spec:
+  entrypoint: run-prediction-classification
+  arguments:
+    parameters:
+      - name: url
+        value: http://sunrise-mlflow-tracking.mlflow:5080
+      - name: input
+        value: iris.csv
+      - name: predictiontype
+        value: classification
+      - name: runid
+        value: 8dbea59123ec469db3ee7b807b3ab6d9
+  templates:
+    - name: run-prediction-classification
+      inputs:
+        parameters:
+          - name: url
+          - name: input
+          - name: predictiontype
+          - name: runid
+        artifacts:
+          - name: file
+            path: /inputfile
+            s3:
+              key: csv/
+      container:
+        image: ppalmes/automlai:v3.0
+        command:
+        args:
+          - "-u{{inputs.parameters.url}}"
+          - "-t{{inputs.parameters.predictiontype}}"
+          - "-p"
+          - "-r{{inputs.parameters.runid}}"
+          - "/inputfile/{{inputs.parameters.input}}"

argo-workflow/automlai-prediction-regression-template.yaml

@@ -0,0 +1,38 @@
+apiVersion: argoproj.io/v1alpha1
+kind: ClusterWorkflowTemplate
+metadata:
+  name: automlai-prediction-regression
+spec:
+  entrypoint: run-prediction-regression
+  arguments:
+    parameters:
+      - name: url
+        value: http://sunrise-mlflow-tracking.mlflow:5080
+      - name: input
+        value: iris_reg.csv
+      - name: predictiontype
+        value: regression
+      - name: runid
+        value: 064fb7a188d34a3da87f2271b8d8d9c2
+  templates:
+    - name: run-prediction-regression
+      inputs:
+        parameters:
+          - name: url
+          - name: input
+          - name: predictiontype
+          - name: runid
+        artifacts:
+          - name: file
+            path: /inputfile
+            s3:
+              key: csv/
+      container:
+        image: ppalmes/automlai:v3.0
+        command:
+        args:
+          - "-u{{inputs.parameters.url}}"
+          - "-t{{inputs.parameters.predictiontype}}"
+          - "-p"
+          - "-r{{inputs.parameters.runid}}"
+          - "/inputfile/{{inputs.parameters.input}}"

argo-workflow/automlai-prediction-template.yaml

@@ -7,13 +7,11 @@ spec:
   arguments:
     parameters:
       - name: url
-        value: http://mlflow.isiath.duckdns.org:8082
+        value: http://sunrise-mlflow-tracking.mlflow:5080
       - name: input
         value: iris_reg.csv
       - name: predictiontype
         value: regression
-      - name: outputloc
-        value: output.txt
       - name: runid
         value: 064fb7a188d34a3da87f2271b8d8d9c2
   templates:
@@ -23,26 +21,18 @@ spec:
           - name: url
           - name: input
           - name: predictiontype
-          - name: outputloc
           - name: runid
         artifacts:
           - name: file
             path: /inputfile
             s3:
               key: csv/
       container:
-        image: ppalmes/automlai:v2.0
+        image: ppalmes/automlai:v3.0
         command:
         args:
-          - "-u {{inputs.parameters.url}}"
-          - "-t {{inputs.parameters.predictiontype}}"
-          - "-o /outputfile"
-          - "--predict_only"
-          - "--runid {{inputs.parameters.runid}}"
+          - "-u{{inputs.parameters.url}}"
+          - "-t{{inputs.parameters.predictiontype}}"
+          - "-p"
+          - "-r{{inputs.parameters.runid}}"
           - "/inputfile/{{inputs.parameters.input}}"
-      outputs:
-        artifacts:
-          - name: ofile
-            path: /outputfile
-            s3:
-              key: "output/{{inputs.parameters.predictiontype}}_{{inputs.parameters.input}}_{{inputs.parameters.outputloc}}.tgz"

argo-workflow/automlai-unisearch-template.yaml

@@ -14,16 +14,10 @@ spec:
         value: regression
       - name: complexity
         value: low
-      - name: outputloc
-        value: NONE
       - name: folds
         value: 3
       - name: workers
         value: 5
-      - name: predict_only
-        value: false
-      - name: runid
-        value: NONE
   templates:
     - name: run-amlp
       inputs:
@@ -32,18 +26,15 @@ spec:
           - name: input
           - name: predictiontype
           - name: complexity
-          - name: outputloc
           - name: folds
           - name: workers
-          - name: predict_only
-          - name: runid
         artifacts:
           - name: file
             path: /inputfile
             s3:
               key: csv/
       container:
-        image: ppalmes/automlai:v2.0
+        image: ppalmes/automlai:v3.0
         command:
         args:
           - "-u{{inputs.parameters.url}}"