MRISegmentation / src /data /data_ingestion.py
smishr-18's picture
Upload 30 files
a578142 verified
raw
history blame contribute delete
734 Bytes
import os
from src import logger
import config.configure as config
print(os.getcwd())
os.environ["KAGGLE_CONFIG_DIR"] = "config/.kaggle/"
import kaggle
class DataIngestion():
def __init__(self):
pass
def download(self):
logger.info(f"Downloading dataset into data/")
dataset_name = config.DATASET_NAME
# Replace '/path/to/your/folder' with the path to the folder where you want to save the dataset
output_folder = config.SAVE_DATA_PATH
# Download the dataset to the specified folder
kaggle.api.dataset_download_files(dataset_name, path=output_folder, unzip=True)
logger.info(f"Download Completed")
if __name__ == '__main__':
data = DataIngestion()