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