Commit
·
ffb59f0
1
Parent(s):
fe82c79
Update README.md
Browse files
README.md
CHANGED
@@ -9,10 +9,7 @@ pipeline_tag: text-classification
|
|
9 |
|
10 |
# librarian-bots/is_new_dataset_student_model
|
11 |
|
12 |
-
This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for text classification. The model
|
13 |
-
|
14 |
-
1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
|
15 |
-
2. Training a classification head with features from the fine-tuned Sentence Transformer.
|
16 |
|
17 |
## Usage
|
18 |
|
@@ -30,11 +27,23 @@ from setfit import SetFitModel
|
|
30 |
# Download from Hub and run inference
|
31 |
model = SetFitModel.from_pretrained("librarian-bots/is_new_dataset_student_model")
|
32 |
# Run inference
|
33 |
-
preds = model([
|
|
|
|
|
|
|
|
|
34 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
## BibTeX entry and citation info
|
37 |
|
|
|
|
|
38 |
```bibtex
|
39 |
@article{https://doi.org/10.48550/arxiv.2209.11055,
|
40 |
doi = {10.48550/ARXIV.2209.11055},
|
|
|
9 |
|
10 |
# librarian-bots/is_new_dataset_student_model
|
11 |
|
12 |
+
This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for text classification. The model is trained to predict whether a title + abstract for a paper on arXiv introduces a new dataset.
|
|
|
|
|
|
|
13 |
|
14 |
## Usage
|
15 |
|
|
|
27 |
# Download from Hub and run inference
|
28 |
model = SetFitModel.from_pretrained("librarian-bots/is_new_dataset_student_model")
|
29 |
# Run inference
|
30 |
+
preds = model([Abstract + Title])
|
31 |
+
```
|
32 |
+
|
33 |
+
During model training, the text was formatted using the following format:
|
34 |
+
|
35 |
```
|
36 |
+
TITLE: title text
|
37 |
+
ABSTRACT: abstract text
|
38 |
+
```
|
39 |
+
|
40 |
+
You probably want to use the same format when running inference for this model.
|
41 |
+
|
42 |
|
43 |
## BibTeX entry and citation info
|
44 |
|
45 |
+
To cite the SetFit approach used to train this model please use this citation:
|
46 |
+
|
47 |
```bibtex
|
48 |
@article{https://doi.org/10.48550/arxiv.2209.11055,
|
49 |
doi = {10.48550/ARXIV.2209.11055},
|