visualization.py

import matplotlib.pyplot as plt

def plot_knowledge(archive):
    fig, ax = plt.subplots()

    if len(archive) > 0:
        ax.plot(archive)
        ax.set_title("Knowledge Field K(t)")
        ax.set_xlabel("Step")
        ax.set_ylabel("K")
    else:
        ax.text(0.5, 0.5, "No Data Yet", ha="center")

    return fig