StankovskiA commited on
Commit
b047db5
1 Parent(s): 3744286

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -0
README.md ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - English
7
+ - RoBERTa-base
8
+ - Text Classification
9
+ pipeline_tag: text-classification
10
+ ---
11
+
12
+ # RoBERTa base Fine-Tuned for Proposal Sentence Classification
13
+
14
+ ## Overview
15
+
16
+ - **Language**: English
17
+ - **Model Name**: oeg/SciBERT-Repository-Proposal
18
+
19
+ ## Description
20
+
21
+ This model is a fine-tuned allenai/scibert_scivocab_uncased model trained to classify sentences into two classes: proposal and non-proposal sentences. The training data includes sentences proposing a software or data repository. The model is trained to recognize and classify these sentences accurately.
22
+
23
+ ## How to use
24
+
25
+ To use this model in Python:
26
+
27
+ ```python
28
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
29
+ import torch
30
+
31
+ tokenizer = AutoTokenizer.from_pretrained("allenai/scibert_scivocab_uncased")
32
+ model = AutoModelForSequenceClassification.from_pretrained("scibert-model")
33
+
34
+ sentence = "Your input sentence here."
35
+ inputs = tokenizer(sentence, return_tensors="pt")
36
+ outputs = model(**inputs)
37
+ probabilities = torch.nn.functional.softmax(outputs.logits, dim=1)