Rajveer Pall commited on
Commit
d6f387a
·
verified ·
1 Parent(s): 62f4371

Upload config.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.py +27 -11
config.py CHANGED
@@ -1,14 +1,30 @@
1
  from pathlib import Path
2
- ROOT_DIR = Path(__file__).resolve().parent
3
- DATA_DIR = ROOT_DIR / "data"
4
- RAW_DIR = DATA_DIR / "raw"
5
- PROCESSED_DIR = DATA_DIR / "processed"
 
 
 
 
 
 
 
 
 
6
  EXPERIMENTS_DIR = ROOT_DIR / "experiments"
7
- START_YEAR = 2018
8
- END_YEAR = 2024
9
- FINBERT_MODEL = "ProsusAI/finbert"
10
- EMBED_MODEL = "all-MiniLM-L6-v2"
11
- BATCH_SIZE = 16
12
- DEVICE = "cpu"
 
 
 
 
 
 
 
13
  TRANSACTION_COST_BPS = 10
14
- REBALANCE_FREQ = "Q"
 
1
  from pathlib import Path
2
+
3
+ # Root
4
+ ROOT_DIR = Path(__file__).resolve().parent
5
+
6
+ # Data paths
7
+ DATA_DIR = ROOT_DIR / "data"
8
+ RAW_DIR = DATA_DIR / "raw"
9
+ PROCESSED_DIR = DATA_DIR / "processed"
10
+
11
+ # Source paths
12
+ SRC_DIR = ROOT_DIR / "src"
13
+
14
+ # Output paths
15
  EXPERIMENTS_DIR = ROOT_DIR / "experiments"
16
+ REPORT_DIR = ROOT_DIR / "report"
17
+
18
+ # Data settings
19
+ START_YEAR = 2018
20
+ END_YEAR = 2024
21
+
22
+ # Model settings
23
+ FINBERT_MODEL = "ProsusAI/finbert"
24
+ EMBED_MODEL = "all-MiniLM-L6-v2"
25
+ BATCH_SIZE = 16
26
+ DEVICE = "cuda"
27
+
28
+ # Backtest settings
29
  TRANSACTION_COST_BPS = 10
30
+ REBALANCE_FREQ = "Q"