Text Classification
Transformers
PyTorch
English
bert
dstc10
knowledge cluster classifier
text-embeddings-inference
Instructions to use wilsontam/bert-base-uncased-dstc10-knowledge-cluster-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use wilsontam/bert-base-uncased-dstc10-knowledge-cluster-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="wilsontam/bert-base-uncased-dstc10-knowledge-cluster-classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("wilsontam/bert-base-uncased-dstc10-knowledge-cluster-classifier") model = AutoModelForSequenceClassification.from_pretrained("wilsontam/bert-base-uncased-dstc10-knowledge-cluster-classifier") - Notebooks
- Google Colab
- Kaggle
This is the model used for knowledge cluster classification for the DSTC10 track2 knowledge selection task, trained with double heads, i.e., classifier head and LM head using ASR error simulator for model training.
For further information, please refer to https://github.com/yctam/dstc10_track2_task2 for the Github repository. You can use this model and use our source code to predict knowledge clusters under ASR errors. AAAI 2022 workshop paper: https://github.com/shanemoon/dstc10/raw/main/papers/dstc10_aaai22_track2_21.pdf
- Downloads last month
- 4