onehot_encode.py

convert labels into onehot format

def encode_onehot(labels: list):
    labels = np.array(labels)
    shape = (labels.size, labels.max()+1)
    one_hot = np.zeros(shape)
    rows = np.arange(labels.size)
    one_hot[rows, labels] = 1
    return one_hot