模型信息

KLAID（韩国法律人工智能数据集）LJP分类模型基于预训练的KLUE RoBERTa-base模型。更多关于KLUE的信息请参考 Github 和 Paper 的详细内容。

如何使用

注意：请使用BertTokenizer替代RobertaTokenizer和RobertaForSequenceClassification。（AutoTokenizer将加载BertTokenizer）

import numpy as np
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("lawcompany/KLAID_LJP_base")
model = AutoModelForSequenceClassification.from_pretrained("lawcompany/KLAID_LJP_base")
model.eval()
input_data = tokenizer("피고인은 2022. 11. 14. 혈중알콜농도 0.123%의 술에 취한 상태로 승용차를 운전하였다.", 
                       max_length=512, 
                       return_tensors="pt")
logits = model(**input_data).logits.detach().numpy()
pred = np.argmax(logits)
# output
# 7

许可信息

版权所有2022至今 Law&Company Co. Ltd.

根据CC-BY-NC-ND-4.0许可

其他咨询

• 电子邮件：klaid@lawcompany.co.kr
• 主页： https://klaid.net/