13 lines
342 B
Plaintext
13 lines
342 B
Plaintext
|
name: s478841 regression model
|
||
|
|
|
||
|
|
docker_env:
|
||
|
|
image: s478841-image:latest
|
||
|
|
|
||
|
|
entry_points:
|
||
|
|
main:
|
||
|
|
parameters:
|
||
|
|
epochs: { type: string, default: "140" }
|
||
|
|
steps: { type: string, default: "10" }
|
||
|
|
save_model: { type: string, default: "--save" }
|
||
|
|
command: "python3 scripts/mlflow_train.py -e {epochs} -s {steps} {save_model}"
|