name lego_sets docker_env: image: s449288/ium:lab8.1 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