11 lines
261 B
Plaintext
11 lines
261 B
Plaintext
|
name: ium_s434766_mlflow_lab8
|
||
|
|
|
||
|
|
docker_env: my_env.yaml
|
||
|
|
image: s434732/ium:latest
|
||
|
|
|
||
|
|
entry_points:
|
||
|
|
main:
|
||
|
|
parameters:
|
||
|
|
epochs = 5
|
||
|
|
batch_size = 10
|
||
|
|
command: "python mlflow.py -r {train_size_param} {test_size_param} {epochs} {batch_size}"
|