Source code for deep_time_series.plotting

import matplotlib.pyplot as plt


def plot_chunks(chunk_specs):
    # ys = [0, 0]
    # widths = [encoding_length, decoding_length]
    # lefts = [0, encoding_length]
    # tags = ['encoding', 'decoding']
    ys = []
    widths = []
    lefts = []
    tags = []

    for y, spec in enumerate(chunk_specs, start=1):
        ys.append(y)
        widths.append(spec.range[1] - spec.range[0])
        lefts.append(spec.range[0])
        tags.append(spec.tag)

    for y, width, left, tag in zip(ys, widths, lefts, tags):
        plt.barh(y=y, width=width, left=left, alpha=0.8)
        plt.annotate(tag, (left, y))