创建精确召回增益曲线并计算曲线下的面积
pyprg的Python项目详细描述
精确召回增益曲线是什么?
内容
此软件包提供以下6个功能:
precision_gain(TP,FN,FP,TN) recall_gain(TP,FN,FP,TN) create_prg_curve(labels,pos_scores) calc_auprg(prg_curve) prg_convex_hull(prg_curve) plot_prg(prg_curve)
安装
可以使用命令行中的pip安装此包:
pip install pyprg
用法
有关用法的详细信息,请参见 单个函数,例如在后面键入?prg.create_prg_curve 使用from pyprg import prg导入。示例用法如下 如下:
frompyprgimportprgimportnumpyasnplabels=np.array([1,1,1,0,1,1,1,1,1,1,0,1,1,1,0,1,0,0,1,0,0,0,1,0,1],dtype='int')scores=np.arange(1,26)[::-1]prg_curve=prg.create_prg_curve(labels,scores,create_crossing_points=True)auprg=prg.calc_auprg(prg_curve)print(auprg)prg.plot_prg(prg_curve)