Python-用TensorFlow实现神经网络实体关系抽取

用TensorFlow实现神经网络实体关系抽取

import sklearn.metrics
import matplotlib
# Use ‘Agg‘ so this program could run on a remote server
matplotlib.use（‘Agg‘）
import matplotlib.pyplot as plt
import numpy as np
import sys
import os

result_dir = ‘./test_result‘

def main（）:
    models = sys.argv[1:]
    for model in models:
        x = np.load（os.path.join（result_dir model +‘_x‘ + ‘.npy‘）） 
        y = np.load（os.path.join（result_dir model + ‘_y‘ + ‘.npy‘））
        f1 = （2 * x * y / （x + y + 1e-20））.max（）
        auc = sklearn.metrics.auc（x=x y=y）
        #plt.plot（x y lw=2 label=model + ‘-auc=‘+str（auc））
        plt.plot（x y lw=2 label=model）
        print（model + ‘ : ‘ + ‘auc = ‘ + str（auc） + ‘ | ‘ + ‘max F1 = ‘ + str（f1））
        print（‘    P@100: {} | P@200: {} | P@300: {} | Mean: {}‘.format（y[100] y[200] y[300] （y[100] + y[200] + y[300]） / 3））
       
    plt.xlabel（‘Recall‘）
    plt.ylabel（‘Precision‘）
    plt.ylim（[0.3 1.0]）
    plt.xlim（[0.0 0.4]）
    plt.title（‘Precision-Recall‘）
    plt.legend（loc=“upper right“）
    plt.grid（True）
    plt.savefig（os.path.join（result_dir ‘pr_curve‘））

if __name__ == “__main__“:
    main（）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2019-08-06 08:08  OpenNRE-master\
     文件         148  2019-08-06 08:08  OpenNRE-master\.gitignore
     文件        1067  2019-08-06 08:08  OpenNRE-master\LICENSE
     文件        7937  2019-08-06 08:08  OpenNRE-master\README.md
     文件           0  2019-08-06 08:08  OpenNRE-master\__init__.py
     文件        1124  2019-08-06 08:08  OpenNRE-master\draw_plot.py
     目录           0  2019-08-06 08:08  OpenNRE-master\nrekit\
     文件          89  2019-08-06 08:08  OpenNRE-master\nrekit\__init__.py
     文件       26761  2019-08-06 08:08  OpenNRE-master\nrekit\data_loader.py
     文件       17671  2019-08-06 08:08  OpenNRE-master\nrekit\framework.py
     目录           0  2019-08-06 08:08  OpenNRE-master\nrekit\network\
     文件          94  2019-08-06 08:08  OpenNRE-master\nrekit\network\__init__.py
     文件        1776  2019-08-06 08:08  OpenNRE-master\nrekit\network\classifier.py
     文件        2268  2019-08-06 08:08  OpenNRE-master\nrekit\network\embedding.py
     文件        3449  2019-08-06 08:08  OpenNRE-master\nrekit\network\encoder.py
     文件        6971  2019-08-06 08:08  OpenNRE-master\nrekit\network\selector.py
     文件       11778  2019-08-06 08:08  OpenNRE-master\nrekit\rl.py
     文件         438  2019-08-06 08:08  OpenNRE-master\requirements.txt
     文件        5377  2019-08-06 08:08  OpenNRE-master\test_demo.py
     文件        6203  2019-08-06 08:08  OpenNRE-master\train_demo.py