File size: 392 Bytes
ca96bc5
 
fcb9629
 
 
ca96bc5
fcb9629
ca96bc5
 
fcb9629
 
1
2
3
4
5
6
7
8
9
10
11
# 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