1
0
mirror of https://github.com/vimagick/dockerfiles.git synced 2024-12-23 01:39:27 +02:00
dockerfiles/ludwig/docker-compose.yml
2019-12-07 04:38:30 +08:00

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