【发布时间】:2019-09-15 17:51:45
【问题描述】:
我正在尝试使用 scikit 学习逻辑回归对多类数据进行分类。我使用一个热编码器对课程进行了编码。但是当我尝试同样的方法时,我得到了错误的输入形状错误。是否可以在 sklearn 逻辑回归中使用一个热编码值?
from sklearn.preprocessing import OneHotEncoder
import numpy as np
from sklearn.linear_model import LogisticRegression
from sklearn.feature_extraction.text import CountVectorizer
onehot_encoder = OneHotEncoder(sparse=False)
y = np.array(y)
ok = onehot_encoder.fit_transform(y.reshape(len(y),1))
vectorizer = CountVectorizer()
X = vectorizer.fit_transform(df2["order_description"])
LogisticRegression().fit(X,ok)
输入: Y - “[0,0,1,0]”
【问题讨论】:
标签: python scikit-learn