subbunanepalli commited on
Commit
1fa293a
·
verified ·
1 Parent(s): fc94fea

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +5 -7
config.py CHANGED
@@ -1,11 +1,9 @@
1
  import os
2
 
3
- # --- Dataset Path ---
4
- BASE_DIR = os.path.abspath(os.path.dirname(__file__))
5
- DATA_PATH = os.path.join(BASE_DIR, "data", "synthetic_transactions_samples_5000.csv")
6
 
7
- # --- Use writable directory ---
8
- MODEL_SAVE_DIR = "/tmp" # ✅ No permission issues
9
  MODEL_PATH = os.path.join(MODEL_SAVE_DIR, "logreg_model.pkl")
10
  TFIDF_VECTORIZER_PATH = os.path.join(MODEL_SAVE_DIR, "tfidf_vectorizer.pkl")
11
  LABEL_ENCODERS_PATH = os.path.join(MODEL_SAVE_DIR, "label_encoders.pkl")
@@ -17,11 +15,11 @@ LABEL_COLUMNS = [
17
  "Risk_Category", "Risk_Drivers", "Investigation_Outcome"
18
  ]
19
 
20
- # --- TF-IDF Settings ---
21
  TFIDF_MAX_FEATURES = 5000
22
  NGRAM_RANGE = (1, 2)
23
  USE_STOPWORDS = True
24
 
25
- # --- Train/Test Split ---
26
  RANDOM_STATE = 42
27
  TEST_SIZE = 0.2
 
1
  import os
2
 
3
+ # --- Paths ---
4
+ MODEL_SAVE_DIR = "/tmp" # Use writable directory
 
5
 
6
+ DATA_PATH = os.path.join(os.path.dirname(__file__), "data", "synthetic_transactions_samples_5000.csv")
 
7
  MODEL_PATH = os.path.join(MODEL_SAVE_DIR, "logreg_model.pkl")
8
  TFIDF_VECTORIZER_PATH = os.path.join(MODEL_SAVE_DIR, "tfidf_vectorizer.pkl")
9
  LABEL_ENCODERS_PATH = os.path.join(MODEL_SAVE_DIR, "label_encoders.pkl")
 
15
  "Risk_Category", "Risk_Drivers", "Investigation_Outcome"
16
  ]
17
 
18
+ # --- TF-IDF ---
19
  TFIDF_MAX_FEATURES = 5000
20
  NGRAM_RANGE = (1, 2)
21
  USE_STOPWORDS = True
22
 
23
+ # --- Split ---
24
  RANDOM_STATE = 42
25
  TEST_SIZE = 0.2