def METHOD_NAME(self, dataset: BigEarthNet) -> None:
    x = dataset[0].copy()
    dataset.plot(x, suptitle="Test")
    plt.close()
    dataset.plot(x, show_titles=False)
    plt.close()
    x["prediction"] = x["label"].clone()
    dataset.plot(x)
    plt.close()