Python中10倍交叉验证代码中的错误

Question

我用Python从头到尾实现了10倍交叉验证。语言是Python3.6，我是用Spyder (Anaconda)写的。我的输入形状是data=(1440,390)，label=(1440,1)。

我的代码：

def partitions(X,y):
  np.random.shuffle(X)
  foldx=[]
  foldy=[]
  j=0
  for i in range(0,10):
    foldx[i]=X[j:j+143,:]
    foldy[i]=y[foldx[j]]
    j+=144
  return np.array(foldx),np.array(foldy)

def cv(X,y,model):
  trainx,trainy=partitions(X,y)
  scores=[]
  for i in range(0,10):
    xtest=trainx[i]
    ytest=trainy[xtest]
    xtrain=trainx[:i]+trainx[i+1:]
    ytrain=trainy[xtrain]
    model.fit(xtrain,ytrain)
    preds=model.predict(xtest)
    print(accuracy_score(np.ravel(ytest),preds))
    scores.append(accuracy_score(np.ravel(ytest),preds))
  return scores.mean()

错误出现在

foldx[i]=X[j:j+143,:]

上面写着

IndexError:超出范围的列表赋值索引。

我该怎么纠正呢？我不是很有经验从零开始解决这些问题。

Answer 1

您必须首先填充一个列表才能使用它的索引，将foldx[i]=X[j:j+143,:]行更改为

 foldx.append(X[j:j+143,:])

同样适用于foldy

foldy.append(y[foldx[j]])