Todos os desafios
Fácil 0 resolvidos 0% de aceitação

Codificação one-hot

Transforme categorias em vetores com um único 1, o formato de entrada de classificadores e embeddings.

O que você vai aprender

  • Mapear categorias para posições de um vetor
  • Montar linhas one-hot para cada rótulo
  • Preparar dados categóricos para modelos

Problema

Ver original (inglês)

Modelos precisam de números, não de palavras. A codificação one-hot transforma cada categoria em um vetor com uma posição por categoria: 1 na posição da categoria e 0 em todas as outras.

Ordene as categorias alfabeticamente. Para cada rótulo, gere seu vetor one-hot.

Implemente one_hot(labels).

Entrada

labels: Uma lista de nomes de categorias.

Saída

Retorne uma matriz com uma linha por rótulo e uma coluna por categoria distinta (em ordem alfabética).

Restrições

1 ≤ comprimento ≤ 1.000

Cada rótulo tem de 1 a 20 caracteres

Exemplos

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

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

Saída 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"};

Saída 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}
};

Explicação

As categorias em ordem alfabética são blue, green, red. Então red → [0, 0, 1], green → [0, 1, 0] e blue → [1, 0, 0].

Exemplo 2
Entrada:
labels = ["yes"]

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

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

Saída esperada:
long[][] expected = new long[][] {
    new long[] {1}
};

Explicação

Há apenas uma categoria, então todo vetor é [1].

Referências

Este desafio foi adaptado para o CodeUtility. Siga o enunciado e as restrições acima.

Aprenda com outros

Soluções da comunidade

Ainda não há soluções publicadas. Resolva este desafio para compartilhar a primeira.