Views
No views yet
1>>> from transformers import pipeline
2>>> unmasker = pipeline('fill-mask', model='andyreas/roberta-gen-news')
3>>> print(unmasker("The weather forecast for <mask> is rain.", top_k=5))
4
5[{'score': 0.06107175350189209,
6'token': 1083,
7'token_str': ' Friday',
8'sequence': 'The weather forecast for Friday is rain.'},
9{'score': 0.04649643227458,
10'token': 1359,
11'token_str': ' Saturday',
12'sequence': 'The weather forecast for Saturday is rain.'
13},
14{'score': 0.04370906576514244,
15'token': 1772,
16'token_str': ' weekend',
17'sequence': 'The weather forecast for weekend is rain.'},
18{'score': 0.04101456701755524,
19'token': 1133,
20'token_str': ' Wednesday',
21'sequence': 'The weather forecast for Wednesday is rain.'},
22{'score': 0.03785591572523117,
23'token': 1234,
24'token_str': ' Sunday',
25'sequence': 'The weather forecast for Sunday is rain.'}]