IUM_04 - update Jenkinsfile and download dataset script

This commit is contained in:
Paweł Łączkowski 2024-03-29 16:45:52 +01:00
parent 157fe812e8
commit 7b0a5e55b9
2 changed files with 4 additions and 11 deletions

11
Jenkinsfile vendored
View File

@ -34,14 +34,11 @@ pipeline {
stage('Download dataset') {
steps {
withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}", "CUTOFF=${params.CUTOFF}"]) {
sh "apt-get install sudo -y"
sh "sudo mkdir /.kaggle"
sh "sudo echo > /.kaggle/kaggle.json"
sh "sudo chmod 777 /.kaggle/kaggle.json"
sh "sudo chown `whoami` /.kaggle/kaggle.json"
sh "kaggle datasets download -d uciml/breast-cancer-wisconsin-data"
sh "unzip -o breast-cancer-wisconsin-data.zip"
sh "mkdir datasets"
sh "mv data.csv datasets/data.csv"
sh "sudo chmod +x ./download_dataset.py"
sh "export KAGGLE_USERNAME=${params.KAGGLE_USERNAME}"
sh "export KAGGLE_KEY=${params.KAGGLE_KEY}"
sh "python3 ./download_dataset.py $CUTOFF"
archiveArtifacts artifacts: './datasets/data.csv,./datasets/train.csv,./datasets/dev.csv,./datasets/test.csv', onlyIfSuccessful: true
}

View File

@ -5,10 +5,6 @@ import kaggle
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import MinMaxScaler
# Download the dataset from Kaggle
kaggle.api.authenticate()
kaggle.api.dataset_download_files('uciml/breast-cancer-wisconsin-data', path='./datasets', unzip=True)
# Load the dataset
df = pd.read_csv('./datasets/data.csv', index_col='id')