Lære OrdinalEncoder

Stryg for at vise menuen

The next issue to address is categorical data. There are two main types of categorical variables.

Ordinal data has a natural order, while nominal data does not. Because of this order, categories can be encoded as numbers according to their ranking.

For example, a 'rate' column with the values 'Terrible', 'Bad', 'OK', 'Good', and 'Great' can be encoded as:

'Terrible' → 0
'Bad' → 1
'OK' → 2
'Good' → 3
'Great' → 4

To encode ordinal data, the OrdinalEncoder is used. It converts categories into integers starting from 0.

OrdinalEncoder is applied in the same way as other transformers. The main challenge lies in specifying the categories argument correctly.

For example, consider a dataset (not the penguins dataset) that contains an 'education' column. The first step is to check its unique values.


              12345
            
import pandas as pd

df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/a65bbc96-309e-4df9-a790-a1eb8c815a1c/adult_edu.csv')

print(df['education'].unique())

An ordered list of categorical values must be created, ranging from 'HS-grad' to 'Doctorate'.


              1234567891011121314
            
import pandas as pd
from sklearn.preprocessing import OrdinalEncoder

# Load the data and assign X, y variables
df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/a65bbc96-309e-4df9-a790-a1eb8c815a1c/adult_edu.csv')
y = df['income'] # 'income' is a target in this dataset
X = df.drop('income', axis=1)
# Create a list of categories so HS-grad is encoded as 0 and Doctorate as 6
edu_categories = ['HS-grad', 'Some-college', 'Assoc', 'Bachelors', 'Masters', 'Prof-school', 'Doctorate']
# Initialize an OrdinalEncoder instance with the correct categories
ord_enc = OrdinalEncoder(categories=[edu_categories])
# Transform the 'education' column and print it
X['education'] = ord_enc.fit_transform(X[['education']])
print(X['education'])

When transforming multiple features with OrdinalEncoder, the categories for each column must be explicitly specified. This is done through the categories argument:

encoder = OrdinalEncoder(categories=[col1_categories, col2_categories, ...])

1. Which statement best describes the use of the `OrdinalEncoder` for handling categorical data in a dataset?

2. Suppose you have a categorical column named `'Color'`. Would it be appropriate to use the `OrdinalEncoder` to encode its values?

Var alt klart?

Tak for dine kommentarer!

Sektion 1. Kapitel 10

Spørg AI

Spørg om hvad som helst eller prøv et af de foreslåede spørgsmål for at starte vores chat

Sektion 1. Kapitel 10

OrdinalEncoder

1. Which statement best describes the use of the OrdinalEncoder for handling categorical data in a dataset?

2. Suppose you have a categorical column named 'Color'. Would it be appropriate to use the OrdinalEncoder to encode its values?

1. Which statement best describes the use of the `OrdinalEncoder` for handling categorical data in a dataset?

2. Suppose you have a categorical column named `'Color'`. Would it be appropriate to use the `OrdinalEncoder` to encode its values?