~/Blog

Brandon Rozek

Photo of Brandon Rozek

PhD Student @ RPI studying Automated Reasoning in AI and Linux Enthusiast.

Reverse One-Hot Encode

Published on

Updated on

Let’s say that you have a dataset that is one hot encoded like the following observation:

import numpy as np
obs = np.array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0,
       0, 0, 0, 0])

The easiest way to reverse one-hot encode the structure, is to take the argmax of the observation.

reverse_encoding = np.argmax(obs)
# 13
Reply via Email Buy me a Coffee
Was this useful? Feel free to share: Hacker News Reddit Twitter