checkpoints/best_model/training_config.json

{
  "PROJECT_ROOT": "D:\\ScamDetectProj\\DetectionMain",
  "TEMP_DIR": "D:\\Temp\\scam_detection",
  "TRANSCRIPT_DIR": "D:\\ScamDetectProj\\DetectionMain\\Transcript",
  "LEGITIMATE_DIR": "D:\\ScamDetectProj\\DetectionMain\\Transcript\\legitimate",
  "SCAM_DIR": "D:\\ScamDetectProj\\DetectionMain\\Transcript\\scam",
  "CHECKPOINT_DIR": "D:\\ScamDetectProj\\DetectionMain\\checkpoints",
  "BEST_MODEL_DIR": "D:\\ScamDetectProj\\DetectionMain\\checkpoints\\best_model",
  "OUTPUTS_DIR": "D:\\ScamDetectProj\\DetectionMain\\outputs",
  "TRAIN_CSV": "D:\\ScamDetectProj\\DetectionMain\\train.csv",
  "VAL_CSV": "D:\\ScamDetectProj\\DetectionMain\\val.csv",
  "TEST_CSV": "D:\\ScamDetectProj\\DetectionMain\\test.csv",
  "TRAIN_SPLIT": 0.8,
  "VAL_SPLIT": 0.1,
  "TEST_SPLIT": 0.1,
  "RANDOM_SEED": 42,
  "MODEL_NAME": "sentence-transformers/all-MiniLM-L6-v2",
  "EMBEDDING_DIM": 384,
  "NUM_LABELS": 2,
  "DROPOUT_RATE": 0.2,
  "MAX_LENGTH": 256,
  "LABEL_MAP": {
    "legitimate": 0,
    "scam": 1
  },
  "ID2LABEL": {
    "0": "LEGITIMATE",
    "1": "SCAM"
  },
  "LEARNING_RATE": 2e-05,
  "WEIGHT_DECAY": 0.01,
  "WARMUP_RATIO": 0.1,
  "NUM_EPOCHS": 10,
  "BATCH_SIZE": 16,
  "GRADIENT_ACCUMULATION_STEPS": 1,
  "MAX_GRAD_NORM": 1.0,
  "EARLY_STOPPING_PATIENCE": 3,
  "EARLY_STOPPING_METRIC": "f1",
  "USE_MIXED_PRECISION": true,
  "ONNX_PATH": "D:\\ScamDetectProj\\DetectionMain\\outputs\\scam_detector.onnx",
  "TORCHSCRIPT_PATH": "D:\\ScamDetectProj\\DetectionMain\\outputs\\scam_detector.pt",
  "HF_API_KEY": "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",
  "max_length": 512
}