Simple visualization of MindSumo challenge data

viz.py

import matplotlib.pyplot as plt
import numpy as np

if __name__ == '__main__':
    input_data = 'data/input_training_0000_0099.npy'
    label_data = 'data/label_training_0000_0099.npy'

    xs = np.load(input_data)
    ys = np.load(label_data)

    fig, (ax1, ax2) = plt.subplots(1,2)
    ax1.imshow(xs[0,5,:,:])
    ax1.set_title('Input temperature')

    ax2.imshow(ys[0,0,:,:])
    ax2.set_title('Label')

    plt.show()