training_metadata.json

{
  "base_model": "sentence-transformers/all-mpnet-base-v2",
  "data_source": "enriched_dataset_with_comments",
  "data_path": "/Users/lovish.agarwal/bwc-analysis-ml/gridsearch/outputs/enriched/flattened_enriched_dataset.json",
  "text_includes": [
    "title",
    "description",
    "comments_concat",
    "investigations_concat",
    "root_cause_concat"
  ],
  "epochs": 3,
  "batch_size": 16,
  "learning_rate": 2e-05,
  "num_iterations": 20,
  "train_test_split": "80/20 stratified",
  "num_samples": 1763,
  "num_test_samples": 441,
  "num_classes": 11,
  "classes": [
    "Accessibility",
    "Availability",
    "Doc",
    "Feature_Request",
    "Gack",
    "Infra",
    "Missed_UseCase",
    "Performance",
    "Regression",
    "ThirdParty",
    "WAD"
  ],
  "training_time_seconds": 60465.716946,
  "seed": 42
}