pipeline {
    agent {
        dockerfile {
            dir 'lab5'
        }
        }
	parameters {
        string(
			defaultValue: '100',
			description: 'Example training parameter',
			name: 'EPOCHS_NUM'
		)
    }
    stages {
        stage('Stage 1') {
            steps {
                    sh 'cd lab5/'
				    sh 'chmod u+x process_dataset.py simple_regression.py'
                    echo 'Processing dataset...'
                    sh 'python3 process_dataset.py'
					echo 'Dataset processed'
					echo 'Conducting simple regression model test'
					sh 'python3 simple_regression.py $EPOCHS_NUM'
					echo 'Model predictions saved'
                    sh 'head lego_reg_results.csv'
				}
            }
        }		
}