我有以下代码
from sklearn.ensemble import ExtraTreesClassifier
from sklearn.cross_validation import cross_val_score
#split the dataset for train and test
combnum['is_train'] = np.random.uniform(0, 1, len(combnum)) <= .75
train, test = combnum[combnum['is_train']==True], combnum[combnum['is_train']==False]
et = ExtraTreesClassifier(n_estimators=200, max_depth=None, min_samples_split=10, random_state=0)
min_samples_split=10, random_state=0 )
labels = train[list(label_columns)].values
tlabels = test[list(label_columns)].values
features = train[list(columns)].values
tfeatures = test[list(columns)].values
et_score = cross_val_score(et, features, labels, n_jobs=-1)
print("{0} -> ET: {1})".format(label_columns, et_score))
检查数组的形状:
features.shape
Out[19]:(43069, 34)
和
labels.shape
Out[20]:(43069, 1)
我得到了:
IndexError: too many indices for array
以及此追溯的相关部分:
---> 22 et_score = cross_val_score(et, features, labels, n_jobs=-1)
我是从Pandas数据框创建数据的,我在这里搜索并通过这种方法看到了一些可能的错误参考,但是无法弄清楚如何纠正? 数据数组的样子: 特征
Out[21]:
array([[ 0., 1., 1., ..., 0., 0., 1.],
[ 0., 1., 1., ..., 0., 0., 1.],
[ 1., 1., 1., ..., 0., 0., 1.],
...,
[ 0., 0., 1., ..., 0., 0., 1.],
[ 0., 0., 1., ..., 0., 0., 1.],
[ 0., 0., 1., ..., 0., 0., 1.]])
标签
Out[22]:
array([[1],
[1],
[1],
...,
[1],
[1],
[1]])
答案 0 :(得分:36)
当我们在scikit-learn中进行交叉验证时,该过程需要(R,)
形状标签而不是(R,1)
。虽然它们在某种程度上是相同的,但它们的索引机制是不同的。所以在你的情况下,只需添加:
c, r = labels.shape
labels = labels.reshape(c,)
在将其传递给交叉验证函数之前。
答案 1 :(得分:14)
如果您将目标标签指定为Pandas中的单个数据列,则似乎可以修复。如果目标有多列,我会收到类似的错误。例如,尝试:
labels = train['Y']
答案 2 :(得分:2)
将.ravel()
添加到传递给公式的Y / Labels变量中也有助于解决KNN中的这个问题。
答案 3 :(得分:0)
尝试目标:
y=df['Survived']
相反,我使用了
y=df[['Survived']]
使目标y成为日期框架,似乎系列会没问题
答案 4 :(得分:0)
您可能需要稍微玩一下尺寸,例如
et_score = cross_val_score(et, features, labels, n_jobs=-1)[:,n]
或
et_score = cross_val_score(et, features, labels, n_jobs=-1)[n,:]
n是维度。