ium_434788/MLproject
Dominik Strzako 1dd2bba636 mlproject
2021-05-16 11:47:17 +02:00

11 lines
277 B
Plaintext

name: 434788-mlflow
docker_env:
image: snowycocoon/ium_434788:4
entry_points:
main:
parameters:
batch_size: {type: int, default: 16}
epochs: {type: int, default: 15}
command: "python3 Zadanie_08_MLflow.py {batch_size} {epochs}"