Team Ai
Modelpublic

adaptive-classifier/document-type

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
2likes803downloads
config.json57 linesDownload Raw Back to root
1{2  "config": {3    "batch_size": 16,4    "cost_coefficients": {},5    "cost_function_type": "separable",6    "device_map": "auto",7    "early_stopping_patience": 3,8    "enable_strategic_mode": false,9    "epochs": 10,10    "ewc_lambda": 100.0,11    "gradient_checkpointing": false,12    "learning_rate": 2e-05,13    "max_examples_per_class": 150,14    "max_length": 512,15    "min_confidence": 0.1,16    "min_examples_per_class": 3,17    "neural_weight": 0.4,18    "num_representative_examples": 5,19    "prototype_update_frequency": 10,20    "prototype_weight": 0.6,21    "quantization": null,22    "similarity_threshold": 0.7,23    "strategic_blend_regular_weight": 0.6,24    "strategic_blend_strategic_weight": 0.4,25    "strategic_lambda": 0.1,26    "strategic_prediction_head_weight": 0.5,27    "strategic_prediction_proto_weight": 0.5,28    "strategic_robust_head_weight": 0.2,29    "strategic_robust_proto_weight": 0.8,30    "strategic_training_frequency": 10,31    "warmup_steps": 032  },33  "embedding_dim": 768,34  "id_to_label": {35    "0": "report",36    "1": "contract",37    "2": "invoice",38    "3": "email",39    "4": "memo"40  },41  "label_to_id": {42    "contract": 1,43    "email": 3,44    "invoice": 2,45    "memo": 4,46    "report": 047  },48  "model_name": "answerdotai/ModernBERT-base",49  "train_steps": 13,50  "training_history": {51    "contract": 100,52    "email": 100,53    "invoice": 100,54    "memo": 100,55    "report": 10056  }57}