diff --git a/modules/util/plot.py b/modules/util/plot.py index 7d2e8376929f54817d3ccb04d7a935907e2b460b..d67c49f0ee1a82f66ca83c7caca602a83f4ad20f 100644 --- a/modules/util/plot.py +++ b/modules/util/plot.py @@ -154,7 +154,7 @@ def plot_roc_curve(correct_labels, prob_pos_class): plt.ylabel('True Positive Rate') plt.title('Receiver operating characteristic') plt.show() - plt.savefig('/home/rink/test_roc', dpi=320) + plt.savefig(homedir+'test_roc', dpi=320) def plot_image(image, fname=None, cmap='Greys', vmin=None, vmax=None):