mirror of
https://github.com/vimagick/dockerfiles.git
synced 2025-01-10 04:19:24 +02:00
27 lines
610 B
YAML
27 lines
610 B
YAML
|
train:
|
||
|
|
image: vimagick/ludwig
|
||
|
|
command: train --data_csv train.csv -mdf model.yaml
|
||
|
|
volumes:
|
||
|
|
- ./data:/data
|
||
|
|
|
||
|
|
visualize:
|
||
|
|
image: vimagick/ludwig
|
||
|
|
command: visualize -v learning_curves -trs results/experiment_run/training_statistics.json -od visualize -ff png
|
||
|
|
volumes:
|
||
|
|
- ./data:/data
|
||
|
|
|
||
|
|
predict:
|
||
|
|
image: vimagick/ludwig
|
||
|
|
command: predict --data_csv predict.csv -m results/experiment_run/model
|
||
|
|
volumes:
|
||
|
|
- ./data:/data
|
||
|
|
|
||
|
|
serve:
|
||
|
|
image: vimagick/ludwig
|
||
|
|
command: serve -m results/experiment_run/model -p 8000
|
||
|
|
ports:
|
||
|
|
- "8000:8000"
|
||
|
|
volumes:
|
||
|
|
- ./data:/data
|
||
|
|
restart: unless-stopped
|