2022-05-13 14:46:06 +02:00
|
|
|
name lego_sets
|
|
|
|
|
|
|
|
|
|
docker_env:
|
2022-05-14 15:49:43 +02:00
|
|
|
image: s449288/ium:lab8.1
|
2022-05-13 14:46:06 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: float, default: 100}
|
|
|
|
|
units: {type: float, default: 1}
|
|
|
|
|
learning_rate: {type: float, default: 0.1}
|
|
|
|
|
command: python3 simple_regression_lab8.py {epochs} {units} {learning_rate}
|
|
|
|
|
test:
|
|
|
|
|
command: python3 evaluate.py
|
