Todos los desafíos
Fácil 0 resueltos 0% de aceptación

Codificación one-hot

Convierte categorías en vectores con un único 1, el formato de entrada de clasificadores y embeddings.

Lo que aprenderás

  • Asociar categorías a posiciones de un vector
  • Construir filas one-hot para cada etiqueta
  • Preparar datos categóricos para modelos

Problema

Ver original (inglés)

Los modelos necesitan números, no palabras. La codificación one-hot convierte cada categoría en un vector con una posición por categoría: 1 en la posición de la categoría y 0 en todas las demás.

Ordena las categorías alfabéticamente. Para cada etiqueta, genera su vector one-hot.

Implementa one_hot(labels).

Entrada

labels: Una lista de nombres de categorías.

Salida

Devuelve una matriz con una fila por etiqueta y una columna por categoría distinta (en orden alfabético).

Restricciones

1 ≤ longitud ≤ 1.000

Cada etiqueta tiene de 1 a 20 caracteres

Ejemplos

Ejemplo 1
Entrada:
labels = ["red", "green", "blue", "green"]

Salida esperada:
expected = [
    [0, 0, 1],
    [0, 1, 0],
    [1, 0, 0],
    [0, 1, 0]
]
Entrada:
std::vector<std::string> labels = {"red", "green", "blue", "green"};

Salida esperada:
std::vector<std::vector<long long>> expected = {
    {0, 0, 1},
    {0, 1, 0},
    {1, 0, 0},
    {0, 1, 0}
};
Entrada:
string[] labels = new string[] {"red", "green", "blue", "green"};

Salida esperada:
long[][] expected = new long[][] {
    new long[] {0, 0, 1},
    new long[] {0, 1, 0},
    new long[] {1, 0, 0},
    new long[] {0, 1, 0}
};

Explicación

Las categorías en orden alfabético son blue, green, red. Así que red → [0, 0, 1], green → [0, 1, 0] y blue → [1, 0, 0].

Ejemplo 2
Entrada:
labels = ["yes"]

Salida esperada:
expected = [
    [1]
]
Entrada:
std::vector<std::string> labels = {"yes"};

Salida esperada:
std::vector<std::vector<long long>> expected = {
    {1}
};
Entrada:
string[] labels = new string[] {"yes"};

Salida esperada:
long[][] expected = new long[][] {
    new long[] {1}
};

Explicación

Solo hay una categoría, así que cada vector es [1].

Referencias

Este reto está adaptado para CodeUtility. Sigue el enunciado y las restricciones indicadas arriba.

Aprende de otros

Soluciones de la comunidad

Aún no hay soluciones publicadas. Resuelve este reto para compartir la primera.