mirror of
https://github.com/vimagick/dockerfiles
synced 2024-06-16 03:48:44 +00:00
27 lines
643 B
YAML
27 lines
643 B
YAML
train:
|
|
image: vimagick/ludwig
|
|
command: train --model_name example --data_csv train.csv -mdf model.yaml
|
|
volumes:
|
|
- ./data:/data
|
|
|
|
visualize:
|
|
image: vimagick/ludwig
|
|
command: visualize -v learning_curves -trs results/experiment_example/training_statistics.json -od visualize -ff png
|
|
volumes:
|
|
- ./data:/data
|
|
|
|
predict:
|
|
image: vimagick/ludwig
|
|
command: predict --data_csv predict.csv -m results/experiment_example/model
|
|
volumes:
|
|
- ./data:/data
|
|
|
|
serve:
|
|
image: vimagick/ludwig
|
|
command: serve -m results/experiment_example/model -p 8000
|
|
ports:
|
|
- "8000:8000"
|
|
volumes:
|
|
- ./data:/data
|
|
restart: unless-stopped
|