ium_434765/MLproject
Karolina Oparczyk 8bb7dab2d8
Some checks failed
s434765-training/pipeline/head There was a failure building this commit
mlflow save model
2021-05-22 17:57:16 +02:00

12 lines
231 B
Plaintext

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