-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain_demo_contextual.py
53 lines (45 loc) · 2.32 KB
/
main_demo_contextual.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from pprint import pprint
from politely import Styler
styler = Styler(scorer="heuristic")
print("##### lm을 쓰지 않는 경우 맥락 고려 X ######")
print(styler("내일 저랑 같이 점심 먹어요.", 0))
pprint(list(sorted(styler.log['guess']['out'], key=lambda x: x[1], reverse=True))[:3])
"""
##### lm을 쓰지 않는 경우 맥락 고려 X ######
내일 나랑 같이 점심 먹어.
[(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '어🏷EF', '.🏷SF'],
0.0125),
(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '대🏷EF', '.🏷SF'], 0.0),
(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '지🏷EF', '.🏷SF'], 0.0)]
"""
styler = Styler(scorer="sbg")
print("##### lm을 쓰는 경우 맥락 고려 O ######")
print(styler("내일 저랑 같이 점심 먹어요.", 0))
pprint(list(sorted(styler.log['guess']['out'], key=lambda x: x[1], reverse=True))[:3])
"""
##### lm을 쓰는 경우 맥락 고려 O ######
내일 나랑 같이 점심 먹자.
[(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '자🏷EF', '.🏷SF'],
-33.544471740722656),
(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '어🏷EF', '.🏷SF'],
-33.63521194458008),
(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '는다🏷EF', '.🏷SF'],
"""
styler = Styler(scorer="gpt2")
print("##### lm을 쓰는 경우 맥락 고려 O ######")
print(styler("내일 저랑 같이 점심 먹어요.", 0))
pprint(list(sorted(styler.log['guess']['out'], key=lambda x: x[1], reverse=True))[:3])
"""
##### lm을 쓰는 경우 맥락 고려 O ######
내일 나랑 같이 점심 먹자.
[(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '자🏷EF', '.🏷SF'],
-6.211542129516602),
(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '어🏷EF', '.🏷SF'],
-6.496356010437012),
(['내일🏷NNG', '나🏷NP', '랑🏷JKB', '같이🏷MAG', '점심🏷NNG', '먹🏷VV', '는다🏷EF', '.🏷SF'],
-6.551154613494873)]
"""
"""
##### lm을 쓰는 경우 맥락 고려 O ######
내일 나랑 같이 점심 먹자.
"""