🚀 芬兰语大小写敏感句子BERT模型
该模型基于FinBERT训练,用于处理芬兰语句子。你可以点击此处查看一个从包含4亿个句子的数据集中检索最相似句子的演示。
🚀 快速开始
本模型可用于计算芬兰语句子之间的相似度,在信息检索、文本匹配等场景有重要应用。
✨ 主要特性
- 语言:芬兰语
- 任务类型:句子相似度计算
- 标签:包含
sentence-transformers
、feature-extraction
、sentence-similarity
、transformers
等标签
- 演示示例:可输入文本 "Minusta täällä on ihana asua!" 进行测试
📦 安装指南
此部分原文档未提及具体安装命令,故跳过。
💻 使用示例
基础用法
from sentence_transformers import SentenceTransformer
sentences = ["Tämä on esimerkkilause.", "Tämä on toinen lause."]
model = SentenceTransformer('TurkuNLP/sbert-cased-finnish-paraphrase')
embeddings = model.encode(sentences)
print(embeddings)
高级用法
from transformers import AutoTokenizer, AutoModel
import torch
def mean_pooling(model_output, attention_mask):
token_embeddings = model_output[0]
input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()
return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)
sentences = ["Tämä on esimerkkilause.", "Tämä on toinen lause."]
tokenizer = AutoTokenizer.from_pretrained('TurkuNLP/sbert-cased-finnish-paraphrase')
model = AutoModel.from_pretrained('TurkuNLP/sbert-cased-finnish-paraphrase')
encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
with torch.no_grad():
model_output = model(**encoded_input)
sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
print("Sentence embeddings:")
print(sentence_embeddings)
📚 详细文档
训练信息
属性 |
详情 |
库 |
sentence-transformers |
FinBERT模型 |
TurkuNLP/bert-base-finnish-cased-v1 |
训练数据 |
此处提供的数据,包括芬兰语释义语料库和自动收集的释义候选(50万个正样本和500万个负样本) |
池化方法 |
均值池化 |
任务类型 |
二分类预测,判断两个句子是否为释义关系。注意:标签3和4被视为释义,标签1和2视为非释义。标签详情 |
模型架构
SentenceTransformer(
(0): Transformer({'max_seq_length': 128, 'do_lower_case': False}) with Transformer model: BertModel
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
)
引用与作者
在相关论文发表之前,请引用此页面。
参考文献
- J. Kanerva, F. Ginter, LH. Chang, I. Rastas, V. Skantsi, J. Kilpeläinen, HM. Kupari, J. Saarni, M. Sevón, and O. Tarkka. Finnish Paraphrase Corpus. In NoDaLiDa 2021, 2021.
- N. Reimers and I. Gurevych. Sentence-BERT: Sentence embeddings using Siamese BERT-networks. In EMNLP-IJCNLP, pages 3982–3992, 2019.
- A. Virtanen, J. Kanerva, R. Ilo, J. Luoma, J. Luotolahti, T. Salakoski, F. Ginter, and S. Pyysalo. Multilingual is not enough: BERT for Finnish. arXiv preprint arXiv:1912.07076, 2019.
🔧 技术细节
原文档未提供具体的技术实现细节,故跳过该章节。
📄 许可证
原文档未提及许可证信息,故跳过该章节。