1
2
mirror of https://github.com/vimagick/dockerfiles synced 2024-06-20 22:08:39 +00:00

update ludwig

This commit is contained in:
kev 2019-12-09 08:33:02 +08:00
parent 190d285681
commit 5195b160d1
4 changed files with 7 additions and 18 deletions

@ -1,14 +1,4 @@
FROM tensorflow/tensorflow:latest-py3 FROM tensorflow/tensorflow:latest-py3
RUN pip install ludwig[full]
RUN apt-get -y install git
RUN git clone --depth=1 https://github.com/uber/ludwig.git \
&& cd ludwig/ \
&& pip install -r requirements.txt -r requirements_text.txt \
-r requirements_image.txt -r requirements_audio.txt \
-r requirements_serve.txt -r requirements_viz.txt \
&& python setup.py install
WORKDIR /data WORKDIR /data
ENTRYPOINT ["ludwig"] ENTRYPOINT ["ludwig"]

@ -48,7 +48,7 @@ $ tree -L 3 data
├── predict.csv ├── predict.csv
├── train.csv ├── train.csv
├── results ├── results
│   └── experiment_run │   └── experiment_example
│   ├── description.json │   ├── description.json
│   ├── model │   ├── model
│   └── training_statistics.json │   └── training_statistics.json

@ -1,24 +1,24 @@
train: train:
image: vimagick/ludwig image: vimagick/ludwig
command: train --data_csv train.csv -mdf model.yaml command: train --model_name example --data_csv train.csv -mdf model.yaml
volumes: volumes:
- ./data:/data - ./data:/data
visualize: visualize:
image: vimagick/ludwig image: vimagick/ludwig
command: visualize -v learning_curves -trs results/experiment_run/training_statistics.json -od visualize -ff png command: visualize -v learning_curves -trs results/experiment_example/training_statistics.json -od visualize -ff png
volumes: volumes:
- ./data:/data - ./data:/data
predict: predict:
image: vimagick/ludwig image: vimagick/ludwig
command: predict --data_csv predict.csv -m results/experiment_run/model command: predict --data_csv predict.csv -m results/experiment_example/model
volumes: volumes:
- ./data:/data - ./data:/data
serve: serve:
image: vimagick/ludwig image: vimagick/ludwig
command: serve -m results/experiment_run/model -p 8000 command: serve -m results/experiment_example/model -p 8000
ports: ports:
- "8000:8000" - "8000:8000"
volumes: volumes:

@ -20,8 +20,7 @@ $ tesseract input.png output -l eng --psm 3
$ cat output.txt $ cat output.txt
The (quick) [brown] {fox} jumps! The (quick) [brown] {fox} jumps!
$ tesseract chinese.jpg chinese -l chi_tra --psm 8 --oem 0 $ tesseract chinese.jpg stdout -l chi_tra --psm 8 --oem 0
$ cat chinese.txt
學習 學習
``` ```