-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
24 lines (18 loc) · 898 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
FROM python:3.7.13-slim
LABEL version="0.2.3"
LABEL image-desc="A docker image for building Propensity models via Vertex AI"
LABEL maintainer="Simeon Thomas"
LABEL org.opencontainers.image.authors="[email protected]"
LABEL org.opencontainers.image.source="https://github.com/olaoluthomas/vertex_prop_train"
LABEL org.opencontainers.image.description="This version expands the list of hyperparameters for SGDClassifier training of lowest segments"
# to show print statements and logs to display in Knative logs
ENV PYTHONUNBUFFERED True
# copy package installer
WORKDIR /
COPY ./dist/vertex_proptrainer-0.3.0-py3-none-any.whl ./
# make RUN commands use `bash --login`
SHELL ["/bin/bash", "--login", "-c"]
# install package
RUN pip install ./vertex_proptrainer-0.0.0-py3-none-any.whl
# establish entrypoint
ENTRYPOINT ["python", "-m", "vertex_proptrainer.train"]