Robust-Biomed-RoBERTa-SemanticSimilarity / prediction_head_0_config.json
mmoradi's picture
Upload prediction_head_0_config.json
d382bf2
{"training": false, "layer_dims": [768, 1], "num_labels": 1, "ph_output_type": "per_sequence", "model_type": "regression", "task_name": "semantic_similarity", "class_weights": null, "pred_threshold": 0.5, "balanced_weights": null, "config": {"training": true, "layer_dims": [768, 1], "num_labels": 1, "ph_output_type": "per_sequence", "model_type": "regression", "task_name": "semantic_similarity", "class_weights": null, "pred_threshold": 0.5, "balanced_weights": null, "name": "RegressionHead"}, "label_tensor_name": "semantic_similarity_label_ids", "label_list": [], "metric": "acc", "name": "RegressionHead"}