Create test.py

test.py

@@ -0,0 +1,59 @@
+import pandas as pd
+import joblib
+from config import TFIDF_VECTORIZER_PATH, MODEL_PATH
+
+# Sample Input Text
+test_text = """
+Transaction ID: T123456789
+Origin: India
+Designation: Manager
+Keywords: sanction, fraud, transaction
+Name: John Doe
+SWIFT Tag: XYZ123
+Currency: USD
+Entity: ACME Corp
+Message: Urgent transaction request
+City: Mumbai
+Country: India
+State: Maharashtra
+Hit Type: Sanctions
+Record Matching String: match-123
+WatchList Match String: sanction-456
+Payment Sender: Sender ABC
+Payment Receiver: Receiver XYZ
+Swift Message Type: MT103
+Text Sanction Data: suspected sanctions
+Matched Sanctioned Entity: Entity123
+Red Flag Reason: Unusual transfer
+Risk Level: High
+Risk Score: 85.5
+CDD Level: Enhanced
+PEP Status: Yes
+Sanction Description: SDN List
+Checker Notes: Needs further review
+Sanction Context: Context of sanction
+Maker Action: Escalated
+Customer Type: Individual
+Industry: Finance
+Transaction Type: Wire
+Transaction Channel: Online
+Geographic Origin: India
+Geographic Destination: Russia
+Risk Category: Category A
+Risk Drivers: Risky geography
+Alert Status: Open
+Investigation Outcome: Pending
+Source of Funds: Unknown
+Purpose of Transaction: Payment
+Beneficial Owner: Jane Roe
+"""
+
+# Load models
+vectorizer = joblib.load(TFIDF_VECTORIZER_PATH)
+model = joblib.load(MODEL_PATH)
+
+# Predict
+test_vec = vectorizer.transform([test_text])
+prediction = model.predict(test_vec)
+
+print("Predicted Labels:", prediction[0])