(12) 자연어처리 RoBERTa를 활용한 NLI실습

from transformers import pipeline, AutoTokenizer

 

classifier = pipeline(

    'text-classification',

    model='Huffon/klue-roberta-base-nli',

    return_all_scores=True

)

tokenizer = AutoTokenizer.from_pretrained('Huffon/klue-roberta-base-nli')

tokenizer.sep_token

classifier(f'나는 악기를 연주하는 것을 좋아해 {tokenizer.sep_token} 나는 음악을 듣고 연주하는 것이 싫어')

classifier(f'여러 남성들이 축구를 하고 있어요. {tokenizer.sep_token} 남자들은 스포츠를 좋아해')