metadata

language:
  - lzh
tags:
  - classical chinese
  - literary chinese
  - ancient chinese
  - token-classification
license: apache-2.0
pipeline_tag: token-classification
widget:
  - text: 子曰學而時習之不亦說乎有朋自遠方來不亦樂乎人不知而不慍不亦君子乎

roberta-classical-chinese-base-sentence-segmentation

Model Description

This is a RoBERTa model pre-trained on Classical Chinese texts for sentence segmentation, derived from roberta-classical-chinese-base-char.

How to Use

import torch
from transformers import AutoTokenizer,AutoModelForTokenClassification
tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-classical-chinese-base-sentence-segmentation")
model=AutoModelForMaskedLM.from_pretrained("KoichiYasuoka/roberta-classical-chinese-base-sentence-segmentation")
s="子曰學而時習之不亦說乎有朋自遠方來不亦樂乎人不知而不慍不亦君子乎"
p=[model.config.id2label[q] for q in torch.argmax(model(tokenizer.encode(s,return_tensors="pt"))[0],dim=2)[0].tolist()[1:-1]]
print("".join(c+"。" if q=="E" or q=="S" else c for c,q in zip(s,p)))