Notice: This page requires JavaScript to function properly.
Please enable JavaScript in your browser settings or update your browser.
Aprenda One-Hot Encoder | Pré-Processamento de Dados com Scikit-Learn
Introdução ao Aprendizado de Máquina com Python

One-Hot Encoder

Deslize para mostrar o menu

Quando se trata de valores nominais, o tratamento é um pouco mais complexo.

Tipos de dados categóricos

Para dados ordinais, como avaliações de usuários que variam de 'Terrível' a 'Ótimo', codificá-los como números de 0 a 4 é adequado, pois o modelo pode capturar a ordem inerente.

Em contraste, para uma característica como 'city' com cinco categorias distintas, codificá-las como números de 0 a 4 sugeriria incorretamente uma ordem. Nesse caso, a codificação one-hot é uma escolha melhor, pois representa as categorias sem implicar uma hierarquia.

Para codificar dados nominais, utiliza-se o transformador OneHotEncoder. Ele cria uma coluna para cada valor único. Em seguida, para cada linha, define 1 na coluna correspondente ao valor dessa linha e 0 nas demais colunas.

OneHotEncoder
Codificação one-hot

O que originalmente era 'NewYork' agora possui 1 na coluna 'City_NewYork' e 0 nas outras colunas City_.

Aplicação do OneHotEncoder ao conjunto de dados penguins. As variáveis nominais são 'island' e 'sex'. A coluna 'species' é o alvo e será tratada separadamente ao abordar codificação do alvo no próximo capítulo.

123456
import pandas as pd df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/a65bbc96-309e-4df9-a790-a1eb8c815a1c/penguins_imputed.csv') print('island: ', df['island'].unique()) print('sex: ', df['sex'].unique())

Para aplicar o OneHotEncoder, inicializar o objeto do codificador e passar as colunas selecionadas para .fit_transform(), da mesma forma que com outros transformadores.

1234567891011
import pandas as pd from sklearn.preprocessing import OneHotEncoder df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/a65bbc96-309e-4df9-a790-a1eb8c815a1c/penguins_imputed.csv') # Assign X, y variables y = df['species'] X = df.drop('species', axis=1) # Initialize an OneHotEncoder object one_hot = OneHotEncoder() # Print transformed 'sex', 'island' columns print(one_hot.fit_transform(X[['sex', 'island']]).toarray())
Note
Nota

O método .toarray() converte a matriz esparsa gerada pelo OneHotEncoder em um array denso do NumPy. Arrays densos exibem todos os valores explicitamente, facilitando a visualização e manipulação dos dados codificados dentro de um DataFrame. Matrizes esparsas armazenam apenas elementos diferentes de zero, otimizando o uso de memória. É possível omitir este método para ver a diferença no resultado.

question mark

OneHotEncoder cria novas colunas. Isso está correto?

Selecione a resposta correta

Tudo estava claro?

Como podemos melhorá-lo?

Obrigado pelo seu feedback!

Seção 2. Capítulo 6

Pergunte à IA

expand

Pergunte à IA

ChatGPT

Pergunte o que quiser ou experimente uma das perguntas sugeridas para iniciar nosso bate-papo

Seção 2. Capítulo 6
some-alt