def one_hot_encode()

in mnist-vscode-docs-sample/utils.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

def one_hot_encode(array, num_of_classes):
    return np.eye(num_of_classes)[array.reshape(-1)]