Skip to content

Commit 071cf61

Browse files
committed
automlai for prediction only
1 parent 3a825ac commit 071cf61

File tree

5 files changed

+53
-5
lines changed

5 files changed

+53
-5
lines changed

DockerizedAutoML/run.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,6 @@ docker run -it --rm --platform=linux/amd64 automlai:v2.0
1111
julia --project -- ./main.jl -t regression --predict_only -u http://mlflow.isiath.duckdns.org:8082 --runid 064fb7a188d34a3da87f2271b8d8d9c2 -o /tmp/reg.txt ./iris_reg.csv
1212
julia --project -- ./main.jl -u http://mlflow.isiath.duckdns.org:8082 -t classification --predict_only --runid e33bbd5c12a54756b1333df1f23a8366 -o /tmp/class.txt ./iris.csv
1313

14-
docker run -it --rm -v `pwd`:/data/ localhost/automlai -u http://mlflow.isiath.duckdns.org:8082 -t classification --predict_only --runid e33bbd5c12a54756b1333df1f23a8366 /data/iris.csv
14+
docker run -it --rm -v `pwd`:/data/ localhost/automlai:v2.0 -u http://mlflow.isiath.duckdns.org:8082 -t classification --predict_only --runid e33bbd5c12a54756b1333df1f23a8366 /data/iris.csv
1515

16-
docker run -it --rm -v `pwd`:/data/ localhost/automlai -u http://mlflow.isiath.duckdns.org:8082 -t regression --predict_only --runid 064fb7a188d34a3da87f2271b8d8d9c2 /data/iris_reg.csv
16+
docker run -it --rm -v `pwd`:/data/ localhost/automlai:v2.0 -u http://mlflow.isiath.duckdns.org:8082 -t regression --predict_only --runid 064fb7a188d34a3da87f2271b8d8d9c2 /data/iris_reg.csv

argo-workflow/automlad-ensemble-template.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
apiVersion: argoproj.io/v1alpha1
22
kind: ClusterWorkflowTemplate
33
metadata:
4-
name: automlad-template
4+
name: automlad
55
spec:
66
entrypoint: run-automlad
77
arguments:

argo-workflow/automlai-dualsearch-template.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ spec:
8383
s3:
8484
key: csv/
8585
container:
86-
image: ppalmes/automlai:v1.0
86+
image: ppalmes/automlai:v2.0
8787
command:
8888
args:
8989
- "-t{{inputs.parameters.predictiontype}}"
Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
apiVersion: argoproj.io/v1alpha1
2+
kind: ClusterWorkflowTemplate
3+
metadata:
4+
name: automlai-prediction
5+
spec:
6+
entrypoint: run-amlp
7+
arguments:
8+
parameters:
9+
- name: url
10+
value: http://sunrise-mlflow-tracking.mlflow:5080
11+
- name: input
12+
value: iris_reg.csv
13+
- name: predictiontype
14+
value: regression
15+
- name: outputloc
16+
value: output.txt
17+
- name: runid
18+
value: NONE
19+
templates:
20+
- name: run-amlp
21+
inputs:
22+
parameters:
23+
- name: url
24+
- name: input
25+
- name: predictiontype
26+
- name: outputloc
27+
- name: runid
28+
artifacts:
29+
- name: file
30+
path: /inputfile
31+
s3:
32+
key: csv/
33+
container:
34+
image: ppalmes/automlai:v2.0
35+
command:
36+
args:
37+
- "-u{{inputs.parameters.url}}"
38+
- "-t{{inputs.parameters.predictiontype}}"
39+
- "-o/outputfile"
40+
- "--predict_only"
41+
- "--runid {{inputs.parameters.runid}}"
42+
- "/inputfile/{{inputs.parameters.input}}"
43+
outputs:
44+
artifacts:
45+
- name: ofile
46+
path: /outputfile
47+
s3:
48+
key: "output/{{inputs.parameters.predictiontype}}_{{inputs.parameters.input}}.tgz"

argo-workflow/automlai-unisearch-template.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ spec:
4343
s3:
4444
key: csv/
4545
container:
46-
image: ppalmes/automlai:v1.0
46+
image: ppalmes/automlai:v2.0
4747
command:
4848
args:
4949
- "-u{{inputs.parameters.url}}"

0 commit comments

Comments
 (0)