logo

[nlg] transformers 라이브러리

 

설치

!pip install transformers
 

간단히 써보기

from transformers import pipeline
classifier = pipeline('sentiment-analysis')
classifier('I am glad to hear that you finally made it.')
[{'label': 'POSITIVE', 'score': 0.9998080134391785}]
classifier('I am sorry that you hate it.')
[{'label': 'NEGATIVE', 'score': 0.9981151223182678}]
 

한국어에 학습된 모형을 사용해보기

classifier = pipeline('sentiment-analysis', model='monologg/koelectra-small-finetuned-nsmc')
classifier('이 영화 진짜 재밌다')
[{'label': 'positive', 'score': 0.9803304076194763}]
classifier('이야기가 말이 안된다')
[{'label': 'negative', 'score': 0.8447015881538391}]
Previous
텐서플로와 파이토치 설치