Skip to content

Commit

Permalink
improve plot
Browse files Browse the repository at this point in the history
  • Loading branch information
zzhanghub committed Dec 12, 2020
1 parent 1754ccd commit 7dd2383
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions plot_curve.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,8 +25,8 @@ def main(cfg):
os.path.join(cfg.res_dir, dataset + '_' + method + '.pth'))
imax = np.argmax(iRes['Fm'])
plt.plot(
iRes['Prec'],
iRes['Recall'],
iRes['Prec'],
# styles[idx_style],
color=colors[idx_style],
linestyle=lines[idx_style],
Expand Down Expand Up @@ -170,4 +170,4 @@ def main(cfg):
parser.add_argument('--res_dir', type=str, default='./')
parser.add_argument('--out_dir', type=str, default=None)
config = parser.parse_args()
main(config)
main(config)

0 comments on commit 7dd2383

Please sign in to comment.