librarian-bot commited on
Commit
ffb59f0
1 Parent(s): fe82c79

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -5
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 has been trained using an efficient few-shot learning technique that involves:
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(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
 
 
 
 
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},