Dockerfile

This commit is contained in:
Mateusz 2024-04-01 17:28:32 +02:00
parent 19f91d4861
commit 545a56f095
1 changed files with 1 additions and 2 deletions

View File

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