Function plot_from_logdir

tsExperiments/scripts_plot/plotting.py:344–414

(logdir, fname, plot_p, main_color, mean_color, rows, cols, dataset, seed=None)

        return os.path.join(logdir, runs[-1])

def plot_from_logdir(logdir, fname, plot_p, main_color, mean_color, rows, cols, dataset, seed=None):

    # Load the data from pickle files
    try:    
        with open(f"{logdir}/context_points.pkl", "rb") as f:
            context_points = pickle.load(f)
    except:
        print(f"No context_points.pkl found in {logdir}")
        logdir = None
        return

    try:
        with open(f"{logdir}/forecast_length.pkl", "rb") as f:
            prediction_length = pickle.load(f)
    except:
        print(f"No forecast_length.pkl found in {logdir}")
        logdir = None
        return

    try:
        with open(f"{logdir}/target_df.pkl", "rb") as f:
            target_df = pickle.load(f)
    except:
        print(f"No target_df.pkl found in {logdir}")
        logdir = None
        return

    try:
        with open(f"{logdir}/hypothesis_forecasts.pkl", "rb") as f:
            hypothesis_forecasts = pickle.load(f)
    except:
        print(f"No hypothesis_forecasts.pkl found in {logdir}")
        logdir = None
        return

    try:
        with open(f"{logdir}/forecast_length.pkl", "rb") as f:
            forecast_length = pickle.load(f)
    except:
        print(f"No forecast_length.pkl found in {logdir}")
        logdir = None
        return

    try:
        with open(f"{logdir}/freq_type.pkl", "rb") as f:
            freq_type = pickle.load(f)
    except:
        print(f"No freq_type.pkl found in {logdir}")
        logdir = None
        return

    try:
        with open(f"{logdir}/is_mcl.pkl", "rb") as f:
            is_mcl = pickle.load(f)
    except:
        print(f"No is_mcl.pkl found in {logdir}")
        logdir = None
        return