def process_iris_labels(labels, operation):
    if (operation == 0):
        p_labels = np.zeros((labels.shape[0], 3))
        for n in range(len(labels)):
            p_labels[(n, int(labels[n]))] = 1
        return p_labels
    else:
        p_labels = np.argmax(labels, axis=1)
        return p_labels
def process_iris_labels(labels, operation):
    if (operation == 0):
        p_labels = np.zeros((labels.shape[0], 3))
        for n in range(len(labels)):
            p_labels[(n, int(labels[n]))] = 1
        return p_labels
    else:
        p_labels = np.argmax(labels, axis=1)
        return p_labels