12 lines
324 B
Plaintext
12 lines
324 B
Plaintext
|
name: 434804
|
||
|
|
|
||
|
|
docker_env: my_env.yaml
|
||
|
|
image: s434804/ium:0.4
|
||
|
|
|
||
|
|
entry_points:
|
||
|
|
main:
|
||
|
|
parameters:
|
||
|
|
test_size: {type: int, default: 0.2}
|
||
|
|
epochs: {type: int, default: 100}
|
||
|
|
batch_size: {type: int, default: 32}
|
||
|
|
command: "python3 mlflow.py {test_size} {epochs} {batch_size}"
|