Aphasia_Classification / config.json
Ellie5757575757's picture
Update config.json
0c96a6d verified
raw
history blame
1.78 kB
{
"architectures": [
"BertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_dropout": null,
"gradient_checkpointing": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 768,
"initializer_range": 0.02,
"intermediate_size": 3072,
"layer_norm_eps": 1e-12,
"max_position_embeddings": 512,
"model_type": "bert",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"pad_token_id": 0,
"position_embedding_type": "absolute",
"transformers_version": "4.30.0",
"type_vocab_size": 2,
"use_cache": true,
"vocab_size": 28996,
"torch_dtype": "float32",
"id2label": {
"0": "Normal",
"1": "Mild Aphasia",
"2": "Moderate Aphasia",
"3": "Severe Aphasia"
},
"label2id": {
"Normal": 0,
"Mild Aphasia": 1,
"Moderate Aphasia": 2,
"Severe Aphasia": 3
},
"num_labels": 4,
"problem_type": "single_label_classification",
"finetuning_task": "aphasia_classification",
"_name_or_path": "dmis-lab/biobert-base-cased-v1.1",
"base_model": "dmis-lab/biobert-base-cased-v1.1",
"task_specific_params": {
"aphasia_classification": {
"max_length": 512,
"do_lower_case": false,
"early_stopping": true,
"num_beams": 1
}
},
"custom_config": {
"model_version": "1.0",
"training_data": "aphasia_corpus",
"evaluation_metrics": {
"accuracy": 0.85,
"f1_score": 0.83,
"precision": 0.84,
"recall": 0.82
},
"feature_extraction": {
"linguistic_features": true,
"syntactic_complexity": true,
"semantic_coherence": true,
"discourse_markers": true
},
"preprocessing": {
"cha_format": true,
"json_structured": true,
"text_normalization": true
}
}
}