import numpy as np
import torch
import matplotlib.pyplot as plt

if __name__ == "__main__":
    data = np.load('./data/GAF_length.npy')
    tensor = torch.from_numpy(data)
    histogram = torch.histc(tensor, bins=10)

    plt.bar(range(10), histogram)
    plt.xlabel('Digits')
    plt.ylabel('Count')
    plt.title('Digit Distribution')
    plt.show()