ium_444380/MLproject

12 lines
279 B
Plaintext
Raw Normal View History

2022-05-15 20:15:27 +02:00
name: s444380
docker_env:
image: kamilguttmann/ium:mlflow
entry_points:
main:
parameters:
epochs: {type: float, default: 10}
command: "python3 train_model_mlflow.py {epochs}"
test:
command: "python3 evaluate_mlflow.py"