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")# pip install -U transformers accelerate # 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", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
9f595a4
1
Parent(s): 529bea9
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
## Article
|
| 2 |
[Medium article](https://medium.com/@shahrukhx01/multi-task-learning-with-transformers-part-1-multi-prediction-heads-b7001cf014bf)
|
| 3 |
## Demo Notebook
|
|
|
|
| 1 |
+
# A Multi-task learning model with two prediction heads
|
| 2 |
+
* First prediction head classifies between keyword sentences vs statements/questions
|
| 3 |
+
* Second prediction head corresponds to classifier for statements vs questions
|
| 4 |
+
|
| 5 |
## Article
|
| 6 |
[Medium article](https://medium.com/@shahrukhx01/multi-task-learning-with-transformers-part-1-multi-prediction-heads-b7001cf014bf)
|
| 7 |
## Demo Notebook
|