Views
No views yet
1from transformers import pipeline
2pipe = pipeline("text-classification", model="dennlinger/roberta-cls-consec")
3
4pipe("{First paragraph} [SEP] {Second paragraph}")[SEP] token. In our training setup, we had entire paragraphs as samples (or up to 512 tokens across two paragraphs), specifically trained on a Terms of Service data set. Note that this might lead to poor performance on "general" topics, such as news articles or Wikipedia.LABELs:LABEL_0: Two input segments separated by [SEP] do not belong to the same topic.LABEL_1: Two input segments separated by [SEP] do belong to the same topic.