ayush2917's picture
Update src/config.py
fcb9629 verified
# src/config.py
DATA_PATH = "data/raw/ecommerce_dataset.csv"
MODEL_NAME = "distilbert-base-uncased"
HF_MODEL_PATH = "your-username/ecommerce-distilbert" # Replace with your HF Hub repo
LOCAL_MODEL_PATH = "models/bert/distilbert"
LOG_FILE = "logs/app.log"
MAX_LENGTH = 128
BATCH_SIZE = 16
EPOCHS = 3
SAMPLE_FRAC = 0.2 # Sample 20% of data
HF_TOKEN = "your-hf-token" # Hugging Face API token