“Exemplo de Albert pré -criado” Respostas de código

Exemplo de Albert pré -criado


from transformers import AlbertTokenizer, AlbertForQuestionAnswering
import torch

tokenizer = AlbertTokenizer.from_pretrained('albert-base-v2')
model = AlbertForQuestionAnswering.from_pretrained('albert-base-v2')
question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet"
input_dict = tokenizer.encode_plus(question, text, return_tensors='pt')
start_scores, end_scores = model(**input_dict)
Successful Shrike

Exemplo de Albert pré -criado


from transformers import AlbertTokenizer, AlbertForQuestionAnswering
import torch

MODEL_PATH = 'ktrapeznikov/albert-xlarge-v2-squad-v2';

tokenizer = AlbertTokenizer.from_pretrained(MODEL_PATH)
model = AlbertForQuestionAnswering.from_pretrained(MODEL_PATH)

Modern Mongoose

Respostas semelhantes a “Exemplo de Albert pré -criado”

Perguntas semelhantes a “Exemplo de Albert pré -criado”

Procure respostas de código populares por idioma

Procurar outros idiomas de código