mmoradi commited on
Commit
862d952
1 Parent(s): 90adbb9

Upload prediction_head_0_config.json

Browse files
Files changed (1) hide show
  1. prediction_head_0_config.json +1 -0
prediction_head_0_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"training": false, "layer_dims": [768, 5], "num_labels": 5, "ph_output_type": "per_sequence", "model_type": "multilabel_text_classification", "task_name": "text_classification", "class_weights": null, "pred_threshold": 0.5, "balanced_weights": null, "config": {"training": true, "layer_dims": [768, 5], "num_labels": 5, "ph_output_type": "per_sequence", "model_type": "multilabel_text_classification", "task_name": "text_classification", "class_weights": null, "pred_threshold": 0.5, "balanced_weights": null, "name": "MultiLabelTextClassificationHead"}, "label_tensor_name": "text_classification_label_ids", "label_list": ["BACKGROUND", "OBJECTIVE", "METHOD", "RESULT", "CONCLUSION"], "metric": "acc", "name": "MultiLabelTextClassificationHead"}