How can I do one-hot encoding in MATLAB?

11 views (last 30 days)
I would like to perform one-hot encoding on the vector [1 7 10 9 8 6]' with 10 classes (numbers 1 to 10). The resulting 6x10 matrix should have a 1x10 vector in place of each number, with "1" at the position corresponding to the number and "0" at all other positions.

Accepted Answer

MathWorks Support Team
MathWorks Support Team on 13 Sep 2024
Edited: MathWorks Support Team on 13 Sep 2024
This can be done by using a logical operator '==' as follows:
>> vec = [1 7 10 9 8 6]'; % Create the vector>> A = (vec==1:10) % Find the column that corresponds to each vector entry
A = 6×10 logical array 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 1 0 0 0 0
You can learn more about the operator here:

More Answers (0)

Categories

Find more on Get Started with MATLAB in Help Center and File Exchange

Products

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!