首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >培训集中的培训数据

培训集中的培训数据
EN

Stack Overflow用户
提问于 2020-01-31 14:46:48
回答 2查看 46关注 0票数 0

Am,试图使用scikit-学习在我的培训集中训练一个模型,但是得到了这个错误:

代码语言:javascript
运行
复制
 ValueError: Expected 2D array, got 1D array instead: array=[90.  4.].
 Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.

步骤1:将x和y分割成训练和测试集

代码语言:javascript
运行
复制
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X,y, test_size = 0.4, random_state = 4)

检查新分割的x值的形状(培训和测试)

代码语言:javascript
运行
复制
X_train = X_train.shape
X_test = X_test.shape
print(X_train)
print(X_test)

检查新分割y值的形状(培训和测试)

代码语言:javascript
运行
复制
y_train = y_train.shape
y_test = y_test.shape
print(y_train)
print(y_test)

第二步:在培训集上培训我们的模型(使用物流回归)

代码语言:javascript
运行
复制
logR = LogisticRegression()
logR = logR.fit(X_train, y_train)

运行这段代码我得到了错误

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2020-01-31 14:54:58

似乎您正在用数据点的形状替换数据点:

代码语言:javascript
运行
复制
X_train = X_train.shape
X_test = X_test.shape
y_train = y_train.shape
y_test = y_test.shape

删除这些行并重新运行。

票数 0
EN

Stack Overflow用户

发布于 2020-01-31 15:04:06

你做了很棒的工作,但是你做了一件错误的事情:你用它们的形状来替换训练和测试数据,这就是你面临这个错误的原因

代码语言:javascript
运行
复制
 #replace these line

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X,y, test_size = 0.4, random_state = 4)

print( X_train.shape)
print( X_test.shape)
print(y_train.shape)
print(y_test.shape)

logR = LogisticRegression()
logR = logR.fit(X_train, y_train)

# Now it work fine
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/60006087

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档