Mizuiro-sakura
commited on
Commit
•
30cd0bc
1
Parent(s):
ea2c349
Upload Mymodel_luke_pn.py
Browse files- Mymodel_luke_pn.py +24 -0
Mymodel_luke_pn.py
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from transformers import MLukeTokenizer
|
3 |
+
from torch import nn
|
4 |
+
|
5 |
+
tokenizer = MLukeTokenizer.from_pretrained('studio-ousia/luke-japanese-base-lite')
|
6 |
+
model = torch.load('C:\\Users\\tomot\\desktop\\Python\\My_luke_model_pn.pth')
|
7 |
+
|
8 |
+
text=input()
|
9 |
+
|
10 |
+
encoded_dict = tokenizer.encode_plus(
|
11 |
+
text,
|
12 |
+
return_attention_mask = True, # Attention maksの作成
|
13 |
+
return_tensors = 'pt', # Pytorch tensorsで返す
|
14 |
+
)
|
15 |
+
|
16 |
+
pre = model(encoded_dict['input_ids'], token_type_ids=None, attention_mask=encoded_dict['attention_mask'])
|
17 |
+
SOFTMAX=nn.Softmax(dim=0)
|
18 |
+
num=SOFTMAX(pre.logits[0])
|
19 |
+
if num[1]>0.5:
|
20 |
+
print(str(num[1]))
|
21 |
+
print('ポジティブ')
|
22 |
+
else:
|
23 |
+
print(str(num[1]))
|
24 |
+
print('ネガティブ')
|