ayush2917 commited on
Commit
fcb9629
·
verified ·
1 Parent(s): 9627370

Update src/config.py

Browse files
Files changed (1) hide show
  1. src/config.py +6 -3
src/config.py CHANGED
@@ -1,8 +1,11 @@
1
  # src/config.py
2
  DATA_PATH = "data/raw/ecommerce_dataset.csv"
3
- MODEL_PATH = "models/bert/distilbert"
 
 
4
  LOG_FILE = "logs/app.log"
5
- MAX_LENGTH = 128 # Max token length for DistilBERT
6
  BATCH_SIZE = 16
7
  EPOCHS = 3
8
- SAMPLE_FRAC = 0.2 # Use 20% of data if too large
 
 
1
  # src/config.py
2
  DATA_PATH = "data/raw/ecommerce_dataset.csv"
3
+ MODEL_NAME = "distilbert-base-uncased"
4
+ HF_MODEL_PATH = "your-username/ecommerce-distilbert" # Replace with your HF Hub repo
5
+ LOCAL_MODEL_PATH = "models/bert/distilbert"
6
  LOG_FILE = "logs/app.log"
7
+ MAX_LENGTH = 128
8
  BATCH_SIZE = 16
9
  EPOCHS = 3
10
+ SAMPLE_FRAC = 0.2 # Sample 20% of data
11
+ HF_TOKEN = "your-hf-token" # Hugging Face API token