ium_434765/MLproject
Karolina Oparczyk d46f43b602
All checks were successful
s434765-training/pipeline/head This commit looks good
s434765-evaluation/pipeline/head This commit looks good
mlflow save model
2021-05-22 17:53:47 +02:00

12 lines
231 B
Plaintext

name: tutorial
docker_env:
image: karopa/ium:30
entry_points:
main:
parameters:
epochs: {type: float, default: 30}
command: "python3 neural_network.py {epochs}"
test:
command: "python3 evaluate_network.py"