ium_470607/lab5/train/MLproject
2021-05-16 23:01:34 +02:00

8 lines
228 B
Plaintext

name: s470607
docker_env:
image: kubakonieczny/ium:train-sacred-mlflow
entry_points:
main:
parameters:
learning_rate: (type: float, default: 0.0003)
comment: "python3 train.py {learning_rate}"