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