mirror of
https://github.com/Hestia-Homes/ML.git
synced 2026-06-08 11:17:25 +00:00
add docker file for testing prediction
This commit is contained in:
parent
f76c238ac6
commit
fbdf4969a6
1 changed files with 14 additions and 0 deletions
14
modules/ml-pipeline/src/pipeline/Prediction.Dockerfile
Normal file
14
modules/ml-pipeline/src/pipeline/Prediction.Dockerfile
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
# Dockerfile that can be used to test loading a model to generate a prediction (part of CI/CD flow)
|
||||
FROM python:3.10.12-slim
|
||||
|
||||
COPY src/requirements/prediction/requirements-dev.txt requirements.txt
|
||||
|
||||
RUN pip install --upgrade pip
|
||||
RUN pip install -r requirements.txt
|
||||
|
||||
# Assuming in the CI/CD step, there will be a dvc pull step to get data and model, so will just need to run a single script
|
||||
COPY src/ /home/src/
|
||||
|
||||
WORKDIR /home/src/
|
||||
|
||||
CMD [ "python", "generate_predictions.py"]
|
||||
Loading…
Add table
Reference in a new issue