2022-05-14 11:50:56 +02:00
|
|
|
name: tutorial
|
|
|
|
|
|
|
|
|
|
# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
|
|
|
|
|
|
|
|
|
|
docker_env:
|
|
|
|
|
image: mikolajk/ium:mlflow
|
|
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
2022-05-14 14:50:00 +02:00
|
|
|
epochs: {type: float, default: 3}
|
2022-05-14 11:50:56 +02:00
|
|
|
learning_rate: {type: float, default: 0.1}
|
2022-05-14 14:50:00 +02:00
|
|
|
command: "python ./lab8/trainScript.py {epochs} {learning_rate}"
|
2022-05-15 12:45:33 +02:00
|
|
|
test:
|
|
|
|
|
command: "python ./src/evalScript.py"
|
