s464937/ium_464937
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

docker

Browse Source
This commit is contained in:
Szymon Bartanowicz 2024-05-15 00:27:38 +02:00
parent dc3284677a
commit cb364fee5f
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -2,7 +2,7 @@ FROM ubuntu:latest
RUN apt-get update && apt-get install -y python3-pip unzip coreutils
RUN pip install --no-cache-dir wheel kaggle pandas scikit-learn tensorflow
RUN pip install --no-cache-dir wheel kaggle pandas scikit-learn tensorflow keras
WORKDIR /app

2
predict.py
View File

@ -4,6 +4,8 @@ from sklearn.preprocessing import StandardScaler, OneHotEncoder
from sklearn.compose import ColumnTransformer
from sklearn.pipeline import Pipeline
from sklearn.model_selection import train_test_split
from keras.metrics import MeanSquaredError
loaded_model = tf.keras.models.load_model('powerlifting_model.h5')