s478841/ium_478841
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

MLProject add

Browse Source
This commit is contained in:
MatOgr 2022-05-15 11:23:25 +02:00
parent 83a7308d1c
commit 17379f6457
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
MLProject Normal file
View File

@ -0,0 +1,12 @@
name: s478841 regression model
docker_env:
image: s478841-image:latest
entry_points:
main:
parameters:
epochs: { type: string, default: "140" }
steps: { type: string, default: "10" }
save_model: { type: string, default: "--save" }
command: "python3 scripts/mlflow_train.py -e {epochs} -s {steps} {save_model}"