Examples · Given a dataset with two features, we let the encoder find the unique values per feature and transform the data to a binary one-hot encoding. >>> from sklearn.preprocessing import OneHotEncoder ·

June 26, 2024 - Scikit-learn's OneHotEncoder can handle unknown categories by ignoring them or assigning them to a dedicated column, ensuring the model can still process new data effectively. This example demonstrates how to fit the encoder on the training data and then transform both training and test data, including handling categories that were not present in the training set.

February 15, 2024 - Another common step, when using sklearn is to do the conversion between raw NumPy arrays and Pandas DataFrames. You can either use sklearn.compose.make_column_transformer for this, or implement it manually, using the .get_feature_names_out() method of OneHotEncoder to give you the column names for the new features. Let’s see examples for both of these.

After saving, you can reload the encoder from the file and perform one-hot encoding, as shown in the following example: encoder_file = "trained_color_segment_encoder.pkl" ... As you can see, the loaded encoder is used exactly the same as the ...

What is one hot encoding?Why use one hot encoding?How to read this decision treeHandling high-cardinality categorical featuresWhat does “high cardinality” mean?Why one-hot encoding becomes problematicSmall vs large category exampleBetter alternatives for high-cardinality featuresTarget encodingFrequency/count encodingHash encodingEmbedding layersWhen should you use each approach?Practical recommendationSparse matrices and memory efficiencyWhy one-hot encoding wastes memoryWhat is a sparse matrix?How Sklearn handles sparse outputDense vs sparse intuitionExample 1: Dense output with PandasOu

Examples · Given a dataset with two features, we let the encoder find the unique values per feature and transform the data to a binary one-hot encoding. >>> from sklearn.preprocessing import OneHotEncoder ·

Let’s look at an example of one hot encoding, using Feature-engine’s OneHotEncoder() utilizing the Titanic Dataset.

September 28, 2021 - from sklearn.preprocessing import OneHotEncoder #creating instance of one-hot-encoder encoder = OneHotEncoder(handle_unknown='ignore') #perform one-hot encoding on 'team' column encoder_df = pd.DataFrame(encoder.fit_transform(df[['team']]).toarray()) #merge one-hot encoded columns back with original DataFrame final_df = df.join(encoder_df) #view final df print(final_df) team points 0 1 2 0 A 25 1.0 0.0 0.0 1 A 12 1.0 0.0 0.0 2 B 15 0.0 1.0 0.0 3 B 14 0.0 1.0 0.0 4 B 19 0.0 1.0 0.0 5 B 23 0.0 1.0 0.0 6 C 25 0.0 0.0 1.0 7 C 29 0.0 0.0 1.0

July 2, 2023 - In this example, we create a sample data frame with a categorical variable “Color.” We then initialize the OneHotEncoder and use the fit_transform method to encode the categorical data.

March 28, 2025 - Here’s a simple example using Pandas: import pandas as pd # Sample categorical data data = {'fruit': ['apple', 'orange', 'banana', 'apple']} df = pd.DataFrame(data) # One Hot Encoding using Pandas get_dummies encoded_df = pd.get_dummies(df, ...

Examples · Given a dataset with three features and two samples, we let the encoder find the maximum value per feature and transform the data to a binary one-hot encoding. >>> from sklearn.preprocessing import OneHotEncoder >>> enc = OneHotEncoder() >>> enc.fit([[0, 0, 3], [1, 1, 0], [0, 2, ...

Examples · Given a dataset with two features, we let the encoder find the unique values per feature and transform the data to a binary one-hot encoding. >>> from sklearn.preprocessing import OneHotEncoder >>> enc = OneHotEncoder(handle_unknown='ignore') >>> X = [['Male', 1], ['Female', 3], ...

April 14, 2024 - The OneHotEncoder class takes an array of data and can be used to one-hot encode the data.

August 14, 2019 - In this example, we will use the encoders from the scikit-learn library. Specifically, the LabelEncoder of creating an integer encoding of labels and the OneHotEncoder for creating a one hot encoding of integer encoded values.

Examples · Given a dataset with three features and four samples, we let the encoder find the maximum value per feature and transform the data to a binary one-hot encoding. >>> from sklearn.preprocessing import OneHotEncoder >>> enc = OneHotEncoder() >>> enc.fit([[0, 0, 3], [1, 1, 0], [0, 2, ...

May 1, 2026 - import pandas as pd from sklearn.preprocessing import OneHotEncoder # create example dataframe data = {'color': ['red', 'blue', 'green', 'red', 'blue']} df = pd.DataFrame(data) # create OneHotEncoder object encoder = OneHotEncoder() # fit and transform color column one_hot_array = encoder.fit_transform(df[['color']]).toarray() # create new dataframe from numpy array one_hot_df = pd.DataFrame(one_hot_array, columns=encoder.get_feature_names()) print(one_hot_df)

August 1, 2024 - # Create dummy variables for categorical features ohe = OneHotEncoder(drop='first', sparse_output=False) # Create dataframe with only the columns that require One Hot Encoding categorical_train = X_train[['type_of_meal_plan', 'required_car_...

January 18, 2022 - The purpose of this article is twofold: first, to give a brief overview of one-hot encoding categorical data, and second to demonstrate how to one-hot encode categorical data using scikit-learn’s OneHotEncoder class. In order for categorical data to be utilized by a machine learning algorithm, it must first be encoded into a numerical form. Categorical variables come in 1 of 2 flavors — ordinal or nominal — and each of these cases should be treated differently. Ordinal data has an inherent ordering between the labels that can easily be converted into a numerical form (small=1, medium=2, or large=3 for example), while nominal data does not (if Nike = 1 and Adidas = 2 does that mean Adidas > Nike?).

>>> enc.categories_ [array(['Female', 'Male'], dtype=object), array([1, 2, 3], dtype=object)] >>> enc.transform([['Female', 1], ['Male', 4]]).toarray() array([[1., 0., 1., 0., 0.], [0., 1., 0., 0., 0.]]) >>> enc.inverse_transform([[0, 1, 1, 0, 0], [0, 0, 0, 1, 0]]) array([['Male', 1], [None, 2]], dtype=object) >>> enc.get_feature_names() array(['x0_Female', 'x0_Male', 'x1_1', 'x1_2', 'x1_3'], dtype=object) >>> drop_enc = OneHotEncoder(drop='first').fit(X) >>> drop_enc.categories_ [array(['Female', 'Male'], dtype=object), array([1, 2, 3], dtype=object)] >>> drop_enc.transform([['Female', 1], ['Male', 2]]).toarray() array([[0., 0., 0.], [1., 1., 0.]])