Cacher is the code snippet organizer for pro developers

# Convert to one hot encoding
train_labels = data['labels']
labels = np.zeros((len(train_labels), 2), np.uint8)
labels[train_labels == 0, 0] = 1
labels[train_labels == 1, 1] = 1