Update README.md
Browse files
README.md
CHANGED
@@ -7,4 +7,57 @@ widget:
|
|
7 |
- text: 예전에는 주말마다 극장에 놀러갔는데 요새는 좀 안가는 편이에요 [SEP] 댓글 주제를 분류하세요 [SEP] 시네마
|
8 |
- text: 인천발 KTX와 관련한 송도역 복합환승센터가 사실상 무산, 단순 철도·버스 위주 환승시설로 만들어진다. 이 때문에 인천시의 인천발 KTX 기점에 앵커시설인 복합환승센터를 통한 인근 지역 경제 활성화를 이뤄낸다는 계획의 차질이 불가피하다. [SEP] 경제에 긍정적인 뉴스인가요? [SEP] 아니요
|
9 |
- text: 마지막에는 k팝 공연보고 좋은 추억 남았으면 좋겠네요 [SEP] 욕설이 포함되어있나요? [SEP] 아니요
|
10 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
- text: 예전에는 주말마다 극장에 놀러갔는데 요새는 좀 안가는 편이에요 [SEP] 댓글 주제를 분류하세요 [SEP] 시네마
|
8 |
- text: 인천발 KTX와 관련한 송도역 복합환승센터가 사실상 무산, 단순 철도·버스 위주 환승시설로 만들어진다. 이 때문에 인천시의 인천발 KTX 기점에 앵커시설인 복합환승센터를 통한 인근 지역 경제 활성화를 이뤄낸다는 계획의 차질이 불가피하다. [SEP] 경제에 긍정적인 뉴스인가요? [SEP] 아니요
|
9 |
- text: 마지막에는 k팝 공연보고 좋은 추억 남았으면 좋겠네요 [SEP] 욕설이 포함되어있나요? [SEP] 아니요
|
10 |
+
---
|
11 |
+
|
12 |
+
## 사용 예시
|
13 |
+
```python
|
14 |
+
# Load model directly
|
15 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
16 |
+
|
17 |
+
tokenizer = AutoTokenizer.from_pretrained("iknow-lab/ko-flan-zero-v0-0731")
|
18 |
+
model = AutoModelForSequenceClassification.from_pretrained("iknow-lab/ko-flan-zero-v0-0731")
|
19 |
+
|
20 |
+
def inference(instruction, input, labels):
|
21 |
+
instruction = f"{input} [SEP] {instruction}"
|
22 |
+
inputs = tokenizer([instruction] * len(labels), labels, truncation=True, padding=True, return_tensors="pt")
|
23 |
+
|
24 |
+
scores = model(**inputs).logits.squeeze(1).tolist()
|
25 |
+
output = dict(zip(labels, scores))
|
26 |
+
|
27 |
+
print(instruction, output)
|
28 |
+
|
29 |
+
inference(
|
30 |
+
"문장을 감성분류해주세요",
|
31 |
+
"아 영화 개노잼",
|
32 |
+
["긍정적", "부정적"]
|
33 |
+
)
|
34 |
+
|
35 |
+
inference(
|
36 |
+
"글과 관련된 내용을 만들어주세요",
|
37 |
+
"예전에는 주말마다 극장에 놀러갔는데 요새는 좀 안가는 편이에요",
|
38 |
+
["영화에 관한 글이다", "드라마에 관한 글입니다"]
|
39 |
+
)
|
40 |
+
|
41 |
+
|
42 |
+
inference(
|
43 |
+
"글을 읽고 시장에 미칠 영향을 판단해보세요",
|
44 |
+
"""인천발 KTX와 관련한 송도역 복합환승센터가 사실상 무산, 단순 철도·버스 위주 환승시설로 만들어진다. 이 때문에 인천시의 인천발 KTX 기점에 앵커시설인 복합환승센터를 통한 인근 지역 경제 활성화를 이뤄낸다는 계획의 차질이 불가피하다.
|
45 |
+
|
46 |
+
25일 시에 따르면 연수구 옥련동 104 일대 29만1천725㎡(8만8천평)에 추진 중인 2만8천62가구 규모의 송도역세권구역 도시개발사업과 연계, KTX 송도역 복합환승센터와 상업시설·업무시설 등의 조성을 추진 중이다. """,
|
47 |
+
["긍정", "부정", "중립"]
|
48 |
+
)
|
49 |
+
|
50 |
+
```
|
51 |
+
### 실행 결과
|
52 |
+
```
|
53 |
+
아 영화 개노잼 [SEP] 문장을 감성분류해주세요
|
54 |
+
{'긍정적': -7.878206253051758, '부정적': 50.96009826660156}
|
55 |
+
|
56 |
+
예전에는 주말마다 극장에 놀러갔는데 요새는 좀 안가는 편이에요 [SEP] 글과 관련된 내용을 만들어주세요
|
57 |
+
{'영화에 관한 글이다': 25.37109375, '드라마에 관한 글입니다': -31.869916915893555}
|
58 |
+
|
59 |
+
인천발 KTX와 관련한 송도역 복합환승센터가 사실상 무산, 단순 철도·버스 위주 환승시설로 만들어진다. 이 때문에 인천시의 인천발 KTX 기점에 앵커시설인 복합환승센터를 통한 인근 지역 경제 활성화를 이뤄낸다는 계획의 차질이 불가피하다.
|
60 |
+
|
61 |
+
25일 시에 따르면 연수구 옥련동 104 일대 29만1천725㎡(8만8천평)에 추진 중인 2만8천62가구 규모의 송도역세권구역 도시개발사업과 연계, KTX 송도역 복합환승센터와 상업시설·업무시설 등의 조성을 추진 중이다. [SEP] 글을 읽고 시장에 미칠 영향을 판단해보세요
|
62 |
+
{'긍정': -61.86758804321289, '부정': 23.72732925415039, '중립': -70.4837417602539}
|
63 |
+
```
|