40 Algorithms Every Programmer Should Know

Feature selection

The process of selecting features that are relevant to the context of the problem that we want to solve is called feature selection. It is an essential part of feature engineering.

Once the file is imported, we drop the User ID column, which is used to identify a person and should be excluded when training a model:

dataset = dataset.drop(columns=['User ID'])

Now let's preview the dataset:

dataset.head(5)

The dataset looks like this:

Now, let's look at how we can further process the input dataset.