gpt4 book ai didi

python - "Inconsistent numbers of samples"-scikit-学习

转载 作者:行者123 更新时间:2023-11-30 09:50:51 24 4
gpt4 key购买 nike

我正在学习 Python 机器学习的一些基础知识(scikit - learn),当我尝试实现 K 最近邻算法时,出现错误:ValueError: 发现输入变量的样本数量不一致:[ 426, 143]。我不知道如何处理。
这是我的代码:

from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier
cancer = load_breast_cancer()
X_train, y_train, X_test, y_test = train_test_split(cancer.data,cancer.target,
stratify =
cancer.target,
random_state = 0)
clf = KNeighborsClassifier(n_neighbors = 6)
clf.fit(X_train, y_train)`

最佳答案

train_test_splitX_train、X_test、y_train、y_test 的顺序返回元组

您将返回值分配给了错误的变量,因此您正在拟合训练数据和测试数据,而不是训练数据和训练标签。

应该是

X_train, X_test, y_train, y_test = train_test_split()

关于python - "Inconsistent numbers of samples"-scikit-学习,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45400034/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com