添加模型和数据信息

Author: rui.tao

README.md

@@ -20,35 +20,27 @@ Best acc on val set: 0.986000
 100%|██████████████████████████████████████████████████████████████████████████████████| 16/16 [00:06<00:00,  2.64it/s, acc=0.986]
 Accuracy on test set: 0.986
 ## 测试结果
->>> model.infer({'text': '场照片事后将发给媒体，避免采访时出现混乱，[3]举行婚礼侯佩岑黄伯俊婚纱照2011年4月17日下午2点，70名亲友见 证下，侯佩', 'h': {'pos': (28, 30)}, 't': {'pos': (31, 33)}})
+model.infer({'text': '场照片事后将发给媒体，避免采访时出现混乱，[3]举行婚礼侯佩岑黄伯俊婚纱照2011年4月17日下午2点，70名亲友见 证下，侯佩', 'h': {'pos': (28, 30)}, 't': {'pos': (31, 33)}})
 ('夫妻', 0.9995878338813782)
->>>
->>> model.infer({'text': '及他们的女儿小苹果与汪峰感情纠葛2004年，葛荟婕在欧洲杯期间录制节目时与汪峰相识并相恋，汪峰那首《我如此爱你', 'h': {'pos': (10, 11)}, 't': {'pos': (22, 24)}})
+
+model.infer({'text': '及他们的女儿小苹果与汪峰感情纠葛2004年，葛荟婕在欧洲杯期间录制节目时与汪峰相识并相恋，汪峰那首《我如此爱你', 'h': {'pos': (10, 11)}, 't': {'pos': (22, 24)}})
 ('情侣', 0.9992896318435669)
->>>
->>> model.infer({'text': '14日，彭加木的侄女彭丹凝打通了彭加木儿子彭海的电话，“堂哥已经知道了，他说这些年传得太多，他不相信是真的', 'h': {'pos': (4, 6)}, 't': {'pos': (22, 21)}})
+
+model.infer({'text': '14日，彭加木的侄女彭丹凝打通了彭加木儿子彭海的电话，“堂哥已经知道了，他说这些年传得太多，他不相信是真的', 'h': {'pos': (4, 6)}, 't': {'pos': (22, 21)}})
 ('父母', 0.8954808712005615)
->>>
->>> model.infer({'text': '名旦吴菱仙是位列“同治十三绝”的名旦时小福的弟子，算得梅兰芳的开蒙老师，早年曾搭过梅巧玲的四喜班，旧谊', 'h': {'pos': (2, 4)}, 't': {'pos': (27, 29)}})
-('师生', 0.996309220790863)
 
-# 使用前准备
-1.bert模型下载：在./pretrain/下面放置chinese_wwm_pytorch模型，下载地址：https://github.com/ymcui/Chinese-BERT-wwm
+model.infer({'text': '名旦吴菱仙是位列“同治十三绝”的名旦时小福的弟子，算得梅兰芳的开蒙老师，早年曾搭过梅巧玲的四喜班，旧谊', 'h': {'pos': (2, 4)}, 't': {'pos': (27, 29)}})
+('师生', 0.996309220790863)
+# 使用方式
+1.bert模型下载：在./pretrain/下面放置chinese_wwm_pytorch模型
 2.数据下载：在./benchmark/people-relation/下执行gen.py,生产中文人物关系数据，具体脚本中有说明。
 3.配置环境变量：vim ~/.bash_profile  添加
     # openNRE
     export openNRE=项目位置
-    
-
-# 注意事项
- 如果自己训练了tensorflow 的bert，可以通过https://github.com/huggingface/transformers 里面的convert_bert_original_tf_checkpoint_to_pytorch.py 脚本转换为pytorch版。
-踩坑：
-1.安装tensorflow 2.0，最终用的都是PyTorch模型，但TensorFlow也得安装
-2.构造checkpoint文件
-3.报错：Embedding' object has no attribute 'shape' ，解决：将报错位置assert那几行直接删除
-
 
 
+-----
+以下是原工程内容
 # OpenNRE
 
 OpenNRE is an open-source and extensible toolkit that provides a unified framework to implement relation extraction models. This package is designed for the following groups:

benchmark/people-relation/gen.py

@@ -0,0 +1,52 @@
+# -*- coding:utf-8 -*-
+# Copyright @rui.tao
+import codecs
+import sys
+import pandas as pd
+import numpy as np
+from collections import deque
+import pdb
+import json
+
+    
+def get_pose(text, E_name):
+    result = []
+    pose_1 = text.find(E_name)
+    pose_2 = pose_1+len(E_name)-1
+    result.append(pose_1)
+    result.append(pose_2)
+    return result
+
+# 处理后数据,处理完了再分训练集和验证集
+f = open('people-relation_all.txt', "w+") 
+
+# 原始数据,数据来源：https://github.com/buppt/ChineseNRE
+people_relation = ""
+
+with codecs.open(people_relation,'r','utf-8') as tfc:
+    for lines in tfc:
+        line = lines.split()
+        data = {} 
+        print(line)
+        E1, E2, relation, text = line
+        data["token"] = list(text)
+        h = {}
+        h["name"] = E1
+        h["pos"] = get_pose(text, E1) 
+        data["h"] = h
+        t = {}
+        t["name"] = E2
+        t["pos"] = get_pose(text, E2)
+        data["t"] = t
+        data["relation"] = relation 
+        json_data = json.dumps(data, ensure_ascii=False)
+        f.write(json_data)
+        f.write("\n")
+        #print(json_data)
+        #import sys
+        #sys.exit()
+        
+
+
+
+

benchmark/people-relation/people-relation_rel2id.json

@@ -0,0 +1 @@
+{"父母":0, "夫妻":1, "师生":2, "兄弟姐妹":3, "合作":4, "情侣":5, "祖孙":6 ,"好友":7, "亲戚":8, "同门":9, "上下级":10, "unknown":11}