반응형
hugging face에서 제공하는 transformers pipeline을 통해 model을 zeroshot으로 사용가능.
from transformers import pipeline
classifier = pipeline("zero-shot-classification",
model="facebook/bart-large-mnli")
model에 hugging face에 올려져있는 다양한 모델을 대입하여 사용가능하다.
Hugging Face – The AI community building the future.
The AI community building the future. Build, train and deploy state of the art models powered by the reference open source in natural language processing.
huggingface.co
- facebook/bart-large-mnli
- bert-base-uncased
- joeddav/bart-large-mnli-yahoo-answers
- classifier에 labels , sentence 외에 multi_class option 가능
- 만약 classifier가 하나의 라벨을 찾는 것이 아니라 여러개의 라벨을 추출하는 경우에는 multi_class = True로 지정.
- 각 클래스의 score를 독립적으로 계산한다.
반응형
'AI' 카테고리의 다른 글
| label studio 데이터 셋 생성 (0) | 2021.09.14 |
|---|---|
| Text classification 성능 측정 (0) | 2021.09.14 |
| hugging face pipeline (0) | 2021.09.13 |
| 코사인 유사도 , 워드 임베딩, 워드 투 벡터 (0) | 2020.08.11 |
| 영화 제목으로 유사한 다른 영화 찾기 (0) | 2020.08.11 |