portguard-api / src /data_loader.py
NickPatel17's picture
Upload 15 files
4856467 verified
raw
history blame contribute delete
272 Bytes
import pandas as pd
from src.config import TRAIN_PATH, TEST_PATH
def load_raw_data():
train_df = pd.read_csv(TRAIN_PATH)
test_df = pd.read_csv(TEST_PATH)
print(f"[Load] Train: {train_df.shape} Test: {test_df.shape}")
return train_df, test_df