Mizuiro-sakura commited on
Commit
30cd0bc
1 Parent(s): ea2c349

Upload Mymodel_luke_pn.py

Browse files
Files changed (1) hide show
  1. 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('ネガティブ')