Dockerfile

This commit is contained in:
Mateusz 2024-04-01 14:30:18 +02:00
parent 6592cd02ac
commit 8b190e2bd8

View File

@ -1,5 +1,4 @@
import os import os
from kaggle.api.kaggle_api_extended import KaggleApi
import pandas as pd import pandas as pd
import numpy as np import numpy as np
@ -9,9 +8,9 @@ from sklearn.model_selection import train_test_split
def download_kaggle_dataset(): def download_kaggle_dataset():
kaggle = KaggleApi() os.system("kaggle datasets download -d mlg-ulb/creditcardfraud")
kaggle.authenticate() os.system("unzip creditcardfraud.zip")
kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True) os.system("rm creditcardfraud.zip")
def load_data(name): def load_data(name):