索引错误:索引 50552 超出轴 0 的范围,大小为 50552

问题描述

我正在尝试运行 Predicted_labels:

def test_ensemble_labels(train_data,y,test_data,vector_names,NNeighbours,lower,upper):
    y_pred = []
    for j in range(len(vector_names)):
        y_pred.append(frnn_owa_method(train_data,vector_names[j],NNeighbours[j],upper)[1])
    # Use voting function to obtain the ensembled label - we used mean
    y_pred_res = np.mean(y_pred,axis=0)
    return y_pred_res 

predicted_labels = test_ensemble_labels(train_data,data['Label'],["Vector_d2v"],[19],additive(),additive())

但我收到一个错误

4 frames
/content/frlearn/neighbours/classifiers.py in <listcomp>(.0)
     24     def construct(self,X,y) -> Model:
     25         model: FuzzyRoughEnsemble.Model = super().construct(X,y)
---> 26         Cs = [X[np.where(y == c)] for c in model.classes]
     27         model.upper_approximations = self.upper_approximator and [self.upper_approximator.construct(C) for C in Cs]
     28         co_Cs = [X[np.where(y != c)] for c in model.classes]

IndexError: index 50552 is out of bounds for axis 0 with size 50552

如何解决这个问题?

解决方法

尝试像这样运行循环

for j in range(len(vector_names) - 1):
    # [..Your stuff...]