ium_444517/MLProject
2022-05-11 14:31:24 +02:00

10 lines
370 B
Plaintext

name: s444517_train
docker_env:
image: kambobdocker420/ium:mlflow
entry_points:
main:
parameters:
epochs: {type: int, default: 200}
first_activation_funct: {type: str, default: "relu"}
second_activation_funct: {type: str, default: "softmax"}
command: "python nn_train_mlflow.py {epochs} {first_activation_funct} {second_activation_funct}"