Instructions to use shahrukhx01/bert-multitask-query-classifiers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use shahrukhx01/bert-multitask-query-classifiers with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="shahrukhx01/bert-multitask-query-classifiers")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("shahrukhx01/bert-multitask-query-classifiers") model = AutoModelForSequenceClassification.from_pretrained("shahrukhx01/bert-multitask-query-classifiers") - Notebooks
- Google Colab
- Kaggle
Commit ·
0dec08e
1
Parent(s): 3074480
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# A Multi-task learning model with two prediction heads
|
| 2 |
-
*
|
| 3 |
-
*
|
| 4 |
|
| 5 |
## Scores
|
| 6 |
##### Spaadia SQuaD Test acc: **0.9891**
|
|
|
|
| 1 |
# A Multi-task learning model with two prediction heads
|
| 2 |
+
* One prediction head classifies between keyword sentences vs statements/questions
|
| 3 |
+
* Other prediction head corresponds to classifier for statements vs questions
|
| 4 |
|
| 5 |
## Scores
|
| 6 |
##### Spaadia SQuaD Test acc: **0.9891**
|