from sklearn.preprocessing import OneHotEncoder
时间: 2023-11-21 17:04:19 浏览: 43
As an AI language model, I cannot import libraries like Scikit-learn or NumPy. However, I can explain what the OneHotEncoder function does.
The OneHotEncoder is a preprocessing step in machine learning that transforms categorical data into numerical data. It takes a categorical variable and converts it into a binary vector where each column represents a unique category and each row represents an observation. The value of each element in the binary vector is either 0 or 1, indicating whether the observation belongs to that category.
For example, if we have a categorical variable "color" with three categories {red, green, blue}, the OneHotEncoder will transform it into a binary vector with three columns {red, green, blue} and each row will have a 1 in the corresponding column that represents the color of the observation.
This encoding is useful in many machine learning algorithms that require numerical inputs, such as linear regression, logistic regression, and neural networks.