Downloads · 30 days
16
2% of all-time downloads
UWB-AIR/MQDD-pretrained
MQDD-pretrained is a feature extraction model from UWB-AIR. Use it when you need embeddings to search or compare text. It is set up for transformers. The card lists the license as cc-by-nc-sa-4.0.
This repository publishes pre-trained model for the paper MQDD – Pre-training of Multimodal Question Duplicity Detection for Software Engineering Domain. For more information, see the paper. The Stack Overflow Dataset…
Downloads · 30 days
16
2% of all-time downloads
All-time downloads
1K
Public
Repo size
1.2 GB
Likes
0
Public
Click a slice to open those files.
.bin585 MB · 100%
From the Hugging Face model README
This repository publishes pre-trained model for the paper MQDD – Pre-training of Multimodal Question Duplicity Detection for Software Engineering Domain. For more information, see the paper. The Stack Overflow Datasets (SOD) and Stack Overflow Duplicity Dataset (SODD) presented in the paper can be obtained from our Stack Overflow Dataset repository.
To acquire the fine-tuned model, see UWB-AIR/MQDD-duplicate.
The MQDD model, which is based on a Longformer architecture and is pre-trained on 218.5M training examples. The model was trained using MLM training objective accompanied with our novel Same Post (SP) and Question Answer (QA) learning objectives targeting specifically the duplicate detection task.
The model can be loaded using the following source code snippet:
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("UWB-AIR/MQDD-pretrained")
model = AutoModel.from_pretrained("UWB-AIR/MQDD-pretrained")
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License. http://creativecommons.org/licenses/by-nc-sa/4.0/
For now, please cite the Arxiv paper:
@misc{https://doi.org/10.48550/arxiv.2203.14093,
doi = {10.48550/ARXIV.2203.14093},
url = {https://arxiv.org/abs/2203.14093},
author = {Pašek, Jan and Sido, Jakub and Konopík, Miloslav and Pražák, Ondřej},
title = {MQDD -- Pre-training of Multimodal Question Duplicity Detection for Software Engineering Domain},
publisher = {arXiv},
year = {2022},
copyright = {Creative Commons Attribution Non Commercial Share Alike 4.0 International}
}