subbunanepalli commited on
Commit
ef1a005
·
verified ·
1 Parent(s): c798099

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +24 -0
config.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ # === Paths ===
4
+ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
5
+
6
+ # Data CSV location
7
+ DATA_PATH = os.path.join(BASE_DIR, "data", "synthetic_transactions_samples_5000.csv")
8
+
9
+ # Saved model and vectorizer paths
10
+ MODEL_SAVE_DIR = os.path.join(BASE_DIR, "saved_models")
11
+ MODEL_PATH = os.path.join(MODEL_SAVE_DIR, "logreg_model.pkl")
12
+ TFIDF_PATH = os.path.join(MODEL_SAVE_DIR, "tfidf_vectorizer.pkl")
13
+
14
+ # Input column (text)
15
+ TEXT_COLUMN = "Sanction_Context"
16
+
17
+ # Output label columns
18
+ LABEL_COLUMNS = [ "Red_Flag_Reason",
19
+ "Maker_Action",
20
+ "Escalation_Level",
21
+ "Risk_Category",
22
+ "Risk_Drivers",
23
+ "Investigation_Outcome"
24
+ ]