10 lines
370 B
Plaintext
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}"
|