One-Hot Encode

Problem

Build a one-hot vector of length num_classes with a 1.0 at position index and 0.0 everywhere else.

The signature is one_hot(index, num_classes). Assume 0 <= index < num_classes.

Hint

Emit 1.0 when the running position equals index.

case 1

in: [0,3]

expected: [1,0,0]

case 2

in: [2,4]

expected: [0,0,1,0]

case 3

in: [0,1]

expected: [1]

case 4

in: [1,1]

expected: [0]

case 5

in: [4,5]

expected: [0,0,0,0,1]

Starter codePython

def one_hot(index, num_classes):
    # Your code here
    pass

5 test cases · run your code against them in the editor