dockerfiles/ludwig/docker-compose.yml

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