Function plot_method_column

tsExperiments/scripts_plot/plotting.py:676–687

Plot a single method's results in one column

(axs, target_df, hypothesis_forecasts, forecast_length, context_points, 
                      freq_type, is_mcl, main_color, mean_color, plot_p=False, ax_cbar=None, extract_unique=True, global_min_prob=None, global_max_prob=None, rows=None, cols=None, dims_to_plot=None)

    plt.close()

def plot_method_column(axs, target_df, hypothesis_forecasts, forecast_length, context_points, 
                      freq_type, is_mcl, main_color, mean_color, plot_p=False, ax_cbar=None, extract_unique=True, global_min_prob=None, global_max_prob=None, rows=None, cols=None, dims_to_plot=None):
    """
    Plot a single method's results in one column
    """
    save_path = f"{os.environ['PROJECT_ROOT']}/tsExperiments/logs/plots/{dataset}"
    if not os.path.exists(save_path):
        os.makedirs(save_path)

    handles, labels = plot_mcl(target_df, hypothesis_forecasts, forecast_length, rows=rows, cols=1, plot_mean=True, context_points=context_points, freq_type=freq_type, extract_unique=extract_unique, save_path=save_path, is_mcl=is_mcl, plot_p=plot_p, main_color=main_color, mean_color=mean_color, dataset=dataset, axs=axs, ax_cbar=ax_cbar, global_min_prob=global_min_prob, global_max_prob=global_max_prob, dims_to_plot=dims_to_plot)

    return handles, labels

# Usage example:
methods = ['tactis2', 'timeGrad', 'tempflow','timeMCL']