def make_barrapunto_figures(output_path_prefix, method_name_list, barrapunto_mse, barrapunto_jaccard, barrapunto_k_list): sns.set_style("darkgrid") sns.set_context("paper") translator = get_method_name_to_legend_name_dict() barrapunto_k_list = list(barrapunto_k_list) fig, axes = plt.subplots(1, 2, sharex=True) axes[0].set_title("BarraPunto Comments") axes[1].set_title("BarraPunto Users") plt.locator_params(nbins=8) # Comments for m, method in enumerate(method_name_list): axes[0].set_ylabel("MSE") axes[0].set_xlabel("Lifetime (sec)") axes[0].plot(barrapunto_k_list[1:], handle_nan( barrapunto_mse[method]["comments"].mean(axis=1))[1:], label=translator[method]) # Users for m, method in enumerate(method_name_list): # axes[1].set_ylabel("MSE") axes[1].set_xlabel("Lifetime (sec)") axes[1].plot(barrapunto_k_list[1:], handle_nan( barrapunto_mse[method]["users"].mean(axis=1))[1:], label=translator[method]) axes[1].legend(loc="upper right") # plt.show() plt.savefig(output_path_prefix + "_mse_barrapunto_SNOW" + ".png", format="png") plt.savefig(output_path_prefix + "_mse_barrapunto_SNOW" + ".eps", format="eps")
def make_barrapunto_figures(output_path_prefix, method_name_list, barrapunto_mse, barrapunto_jaccard, barrapunto_k_list): sns.set_style("darkgrid") sns.set_context("paper") translator = get_method_name_to_legend_name_dict() barrapunto_k_list = list(barrapunto_k_list) fig, axes = plt.subplots(1, 2, sharex=True) axes[0].set_title("BarraPunto Comments") axes[1].set_title("BarraPunto Users") plt.locator_params(nbins=8) # Comments for m, method in enumerate(method_name_list): axes[0].set_ylabel("MSE") axes[0].set_xlabel("Lifetime (sec)") axes[0].plot(barrapunto_k_list[1:], handle_nan(barrapunto_mse[method]["comments"].mean(axis=1))[1:], label=translator[method]) # Users for m, method in enumerate(method_name_list): # axes[1].set_ylabel("MSE") axes[1].set_xlabel("Lifetime (sec)") axes[1].plot(barrapunto_k_list[1:], handle_nan(barrapunto_mse[method]["users"].mean(axis=1))[1:], label=translator[method]) axes[1].legend(loc="upper right") # plt.show() plt.savefig(output_path_prefix + "_mse_barrapunto_SNOW" + ".png", format="png") plt.savefig(output_path_prefix + "_mse_barrapunto_SNOW" + ".eps", format="eps")
def make_reddit_figures(output_path_prefix, method_name_list, reddit_mse, reddit_jaccard, reddit_k_list): sns.set_style("darkgrid") sns.set_context("paper") translator = get_method_name_to_legend_name_dict() reddit_k_list = list(reddit_k_list) fig, axes = plt.subplots(1, 2, sharex=True) axes[0].set_title("RedditNews Comments") axes[1].set_title("RedditNews Users") plt.locator_params(nbins=8) # Comments for m, method in enumerate(method_name_list): axes[0].set_ylabel("MSE") axes[0].set_xlabel("Lifetime (sec)") axes[0].plot(reddit_k_list[1:], handle_nan(reddit_mse[method]["comments"].mean(axis=1))[1:], label=translator[method]) # Users for m, method in enumerate(method_name_list): # axes[1].set_ylabel("MSE") axes[1].set_xlabel("Lifetime (sec)") axes[1].plot(reddit_k_list[1:], handle_nan(reddit_mse[method]["users"].mean(axis=1))[1:], label=translator[method]) axes[1].legend(loc="lower left") # plt.show() plt.savefig(output_path_prefix + "_mse_com_use_SNOW" + ".png", format="png") plt.savefig(output_path_prefix + "_mse_com_use_SNOW" + ".eps", format="eps") #################################################################################################################### fig, axes = plt.subplots(1, 2, sharex=True) axes[0].set_title("RedditNews Score") axes[1].set_title("RedditNews Controversiality") plt.locator_params(nbins=8) # Score-Wilson for m, method in enumerate(method_name_list): axes[0].set_ylabel("MSE") axes[0].set_xlabel("Lifetime (sec)") axes[0].plot(reddit_k_list[1:], handle_nan(reddit_mse[method]["score_wilson"].mean(axis=1))[1:], label=translator[method]) # Controversiality - Wilson for m, method in enumerate(method_name_list): # axes[1].set_ylabel("MSE") axes[1].set_xlabel("Lifetime (sec)") axes[1].plot(reddit_k_list[1:], handle_nan(reddit_mse[method]["controversiality_wilson"].mean(axis=1))[1:], label=translator[method]) axes[1].legend(loc="lower left") # plt.show() plt.savefig(output_path_prefix + "_mse_sco_ctr_SNOW" + ".png", format="png") plt.savefig(output_path_prefix + "_mse_sco_ctr_SNOW" + ".eps", format="eps") #################################################################################################################### # Controversiality - Wilson for m, method in enumerate(method_name_list): print(method, "Top-100 Jaccard index") print(reddit_jaccard[method]["controversiality_wilson"].mean(axis=1))