Dockerfile
This commit is contained in:
parent
1f518e4c14
commit
52be040de8
@ -1,4 +1,5 @@
|
|||||||
import os
|
import os
|
||||||
|
import sys
|
||||||
from kaggle.api.kaggle_api_extended import KaggleApi
|
from kaggle.api.kaggle_api_extended import KaggleApi
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import numpy as np
|
import numpy as np
|
||||||
@ -10,6 +11,8 @@ from sklearn.model_selection import train_test_split
|
|||||||
|
|
||||||
def download_kaggle_dataset():
|
def download_kaggle_dataset():
|
||||||
print(f'Kaggle API key: {os.environ["KAGGLE_KEY"]}')
|
print(f'Kaggle API key: {os.environ["KAGGLE_KEY"]}')
|
||||||
|
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)
|
||||||
|
Loading…
Reference in New Issue
Block a user