diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/make_plots.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/make_plots.py b/src/make_plots.py index 85850b1..c45479e 100644 --- a/src/make_plots.py +++ b/src/make_plots.py @@ -71,7 +71,7 @@ def plot_watts_strogatz_graph(): 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="lower right") + plt.legend(loc="lower right", fontsize=18) ax.set_xticks(x) ax.set_xticklabels(tuple(labels)) plt.savefig("../paper/figures/"+"watts_strogatz.pdf") @@ -200,7 +200,7 @@ def plot_ROC_curve(figure_name): plt.savefig("../paper/figures/"+"ROC_curve.pdf") -def plot_kronecker_l2norm_nonsparse(): +def plot_p_init_watts_strogatz(): plt.clf() fig = plt.figure(1) x = [.01, .05, .1, .15, .2] @@ -216,13 +216,13 @@ def plot_kronecker_l2norm_nonsparse(): plt.axis((0, .21, .05, .8)) plt.xlabel("Number of Cascades", fontsize=20) - plt.ylabel("l2-norm", fontsize=20) + plt.ylabel("F1-score", 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", fontsize=18) + plt.legend(loc="upper right", fontsize=16) ax.set_xticks(x) ax.set_yticklabels(tuple([.1, .2, .3, .4, .5, .6, .7, .8]), fontsize=20) ax.set_xticklabels(tuple(x), fontsize=20) |