mlflow
This commit is contained in:
parent
6aa9dfba3d
commit
73a958e269
16
MLproject.yml
Normal file
16
MLproject.yml
Normal file
@ -0,0 +1,16 @@
|
||||
name: s487194
|
||||
|
||||
docker_env:
|
||||
image: s487194/ium:1
|
||||
|
||||
entry_points:
|
||||
main:
|
||||
parameters:
|
||||
epochs: {type: int, default: 100}
|
||||
lr: {type: float, default: 0.01}
|
||||
command: "python train.py --epochs {epochs} --lr {lr}"
|
||||
test:
|
||||
parameters:
|
||||
|
||||
command: "python prediction.py"
|
||||
|
Loading…
Reference in New Issue
Block a user