예제 #1
0
                                       'ro',
                                       markersize=4)
        disapproval_predicted, = plt.plot(axis_vals,
                                          predict_disapproval,
                                          'bo',
                                          markersize=4)
        plt.legend([disapproval_actual, approval_predicted],
                   ["Actual", "Predicted"],
                   loc=2,
                   borderaxespad=0.)
        plt.ylabel('Disapproval percentage')
        axes = plt.gca()
        axes.set_ylim([20, 70])

        plt.show()
        config_params = [
            "CORPUS_NAME", "POLL_DELAY", "DAY_RANGE",
            "SENTIMENT_ANALYSIS_METHOD", "TOPIC_EXTRACTION_METHOD",
            "NUM_TOPICS", "REGRESSION_MODEL", "NUM_LAYERS", "YEARS"
        ]
        plt.savefig(
            os.path.join(Config.PLOT_DIR,
                         (Config.dump_config(config_params) + ".png")))
        pickle.dump(
            k_fold_scores,
            open(
                os.path.join(
                    Config.PLOT_DIR,
                    Config.dump_config(config_params) +
                    "_k_fold_scores_negmse.txt"), "wb"))