12 lines
279 B
Plaintext
12 lines
279 B
Plaintext
|
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"
|