Spaces:
No application file
No application file
Update src/config.py
Browse files- 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 |
-
|
|
|
|
|
|
|
| 4 |
LOG_FILE = "logs/app.log"
|
| 5 |
-
MAX_LENGTH = 128
|
| 6 |
BATCH_SIZE = 16
|
| 7 |
EPOCHS = 3
|
| 8 |
-
SAMPLE_FRAC = 0.2 #
|
|
|
|
|
|
| 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
|