pipeline {
    agent {
        dockerfile true
        }
	parameters {
        string(
			defaultValue: '100',
			description: 'Example training parameter',
			name: 'EPOCHS_NUM'
		)
    }
    stages {
        stage('Stage 1') {
            steps {
				    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'
				}
            }
        }		
}