diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/make_plots.py | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/src/make_plots.py b/src/make_plots.py index c5a50f6..9615d1b 100644 --- a/src/make_plots.py +++ b/src/make_plots.py @@ -193,7 +193,7 @@ def plot_ROC_curve(figure_name): color="orange", label="Lasso-50 cascades") ax.plot(recall_sparse_50, precision_sparse_50, 'ko-', color="cornflowerblue", label="Our Method-50 cascades") - plt.legend(loc="upper right") + plt.legend(loc="upper right", fontsize=14) plt.savefig("../paper/figures/"+"ROC_curve.pdf") @@ -208,20 +208,21 @@ def plot_kronecker_l2norm_nonsparse(): fig, ax = plt.subplots() - #plt.subplots_adjust(bottom=.2, top=.85) - #plt.xticks(ha="right", rotation=45) + plt.subplots_adjust(bottom=.2, top=.85) + plt.xticks(ha="right", rotation=45) - #plt.axis((50, 2000, 0, 145)) - plt.xlabel("Number of Cascades") - plt.ylabel("l2-norm") + plt.axis((0, .21, .05, .8)) + plt.xlabel("Number of Cascades", fontsize=20) + plt.ylabel("l2-norm", fontsize=20) plt.grid(color="lightgrey") ax.plot(x, greedy, 'ko-', color="lightseagreen", label='Greedy') ax.plot(x, lasso, 'ko-', color="orange", label="Lasso") ax.plot(x, max_likel, 'ko-', color="cornflowerblue", label="MLE") ax.plot(x, sparse_recov, 'ko-', color="k", label="Our Method") - plt.legend(loc="upper right") + plt.legend(loc="upper right", fontsize=18) ax.set_xticks(x) - ax.set_xticklabels(tuple(x)) + ax.set_yticklabels(tuple([.1, .2, .3, .4, .5, .6, .7, .8]), fontsize=20) + ax.set_xticklabels(tuple(x), fontsize=20) plt.savefig("../paper/figures/"+"watts_strogatz_p_init.pdf") |