Robust-Biomed-RoBERTa-QuestionAnswering / prediction_head_0_config.json
mmoradi's picture
Upload prediction_head_0_config.json
afbca76
raw history blame
No virus
705 Bytes
{"training": false, "layer_dims": [768, 3], "num_labels": 3, "ph_output_type": "per_sequence", "model_type": "multilabel_text_classification", "task_name": "Question_answering", "class_weights": null, "pred_threshold": 0.5, "balanced_weights": null, "config": {"training": true, "layer_dims": [768, 3], "num_labels": 3, "ph_output_type": "per_sequence", "model_type": "multilabel_text_classification", "task_name": "question_answering", "class_weights": null, "pred_threshold": 0.5, "balanced_weights": null, "name": "MultiLabelTextClassificationHead"}, "label_tensor_name": "question_answering_label_ids", "label_list": ["yes", "no", "maybe"], "metric": "acc", "name": "MultiLabelTextClassificationHead"}