albert-pt-br

作者 josu · 下载 0 · 点赞 1 · 更新 2023-10-31

portuguese brazil pt_BR

from transformers import pipeline, AlbertTokenizer, AlbertForMaskedLM
model = AlbertForMaskedLM.from_pretrained('josu/albert-pt-br')
tokenizer = AlbertTokenizer.from_pretrained('josu/albert-pt-br')
unmasker = pipeline('fill-mask', model=model, tokenizer=tokenizer ,device=0)

text = 'Marte está no [MASK] solar.'
unmasker(text)

[{'score': 0.7004144191741943,
  'token': 244,
  'token_str': 'sistema',
  'sequence': 'marte esta no sistema solar.'},
 {'score': 0.02539917267858982,
  'token': 4077,
  'token_str': 'solar',
  'sequence': 'marte esta no solar solar.'},
 {'score': 0.020301498472690582,
  'token': 49,
  'token_str': 'seu',
  'sequence': 'marte esta no seu solar.'},
 {'score': 0.01753508299589157,
  'token': 482,
  'token_str': 'centro',
  'sequence': 'marte esta no centro solar.'},
 {'score': 0.013344300910830498,
  'token': 1401,
  'token_str': 'plano',
  'sequence': 'marte esta no plano solar.'}]
  

模型信息

该模型卡暂无预览图和文字说明。以下信息来自 Hugging Face。

作者 josu
类型—
基础模型—
标签portuguese、brazil、pt_BR
下载0
点赞1
更新时间2023-10-31
Hugging Face https://huggingface.co/josu/albert-pt-br