Dockerfile

This commit is contained in:
Mateusz 2024-04-01 17:26:36 +02:00
parent 78754f7a40
commit 19f91d4861
2 changed files with 4 additions and 2 deletions

4
Jenkinsfile vendored
View File

@ -28,8 +28,8 @@ pipeline {
stages { stages {
stage('Run create-dataset script') { stage('Run create-dataset script') {
steps { steps {
sh 'chmod +x create-dataset.py' sh 'chmod +x create-dataset.py'
sh 'python3 ./create-dataset.py' sh 'python3 ./create-dataset.py $KAGGLE_USERNAME $KAGGLE_KEY'
} }
} }
stage('Archive Artifacts') { stage('Archive Artifacts') {

View File

@ -10,6 +10,8 @@ from sklearn.model_selection import train_test_split
def download_kaggle_dataset(): def download_kaggle_dataset():
os.environ["KAGGLE_USERNAME"] = sys.argv[1]
os.environ["KAGGLE_KEY"] = sys.argv[2]
kaggle = KaggleApi() kaggle = KaggleApi()
kaggle.authenticate() kaggle.authenticate()
kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True) kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True)