mirror of
https://github.com/vimagick/dockerfiles.git
synced 2024-12-23 01:39:27 +02:00
update ludwig
This commit is contained in:
parent
190d285681
commit
5195b160d1
@ -1,14 +1,4 @@
|
||||
FROM tensorflow/tensorflow:latest-py3
|
||||
|
||||
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
|
||||
|
||||
RUN pip install ludwig[full]
|
||||
WORKDIR /data
|
||||
|
||||
ENTRYPOINT ["ludwig"]
|
||||
|
@ -48,7 +48,7 @@ $ tree -L 3 data
|
||||
├── predict.csv
|
||||
├── train.csv
|
||||
├── results
|
||||
│ └── experiment_run
|
||||
│ └── experiment_example
|
||||
│ ├── description.json
|
||||
│ ├── model
|
||||
│ └── training_statistics.json
|
||||
|
@ -1,24 +1,24 @@
|
||||
train:
|
||||
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:
|
||||
- ./data:/data
|
||||
|
||||
visualize:
|
||||
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:
|
||||
- ./data:/data
|
||||
|
||||
predict:
|
||||
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:
|
||||
- ./data:/data
|
||||
|
||||
serve:
|
||||
image: vimagick/ludwig
|
||||
command: serve -m results/experiment_run/model -p 8000
|
||||
command: serve -m results/experiment_example/model -p 8000
|
||||
ports:
|
||||
- "8000:8000"
|
||||
volumes:
|
||||
|
@ -20,8 +20,7 @@ $ tesseract input.png output -l eng --psm 3
|
||||
$ cat output.txt
|
||||
The (quick) [brown] {fox} jumps!
|
||||
|
||||
$ tesseract chinese.jpg chinese -l chi_tra --psm 8 --oem 0
|
||||
$ cat chinese.txt
|
||||
$ tesseract chinese.jpg stdout -l chi_tra --psm 8 --oem 0
|
||||
學習
|
||||
```
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user