我正在使用GridSearchCV训练Ann机器学习模型，并在gridSearchCV中陷入IndexError问题 [英] I was training an Ann machine learning model using GridSearchCV and got stuck with an IndexError in gridSearchCV

问题描述

我的模型开始训练，执行一段时间后出现错误:- IndexError:索引37超出了尺寸为37的轴0的边界

My model starts to train and while executing for sometime it gives an error :- IndexError: index 37 is out of bounds for axis 0 with size 37

对于我的模型，它可以正确执行，而无需使用带有固定参数的gridsearchCV

It executes properly for my model without using gridsearchCV with fixed parameters

这是我的代码

    from keras.wrappers.scikit_learn import KerasClassifier
    from sklearn.model_selection import GridSearchCV
    from keras.models import Sequential
    from keras.layers import Dense
    def build_classifier(optimizer, nb_layers,unit):
        classifier = Sequential()
        classifier.add(Dense(units = unit, kernel_initializer = 'uniform', activation = 'relu', input_dim = 14))
        i = 1
        while i <= nb_layers:
            classifier.add(Dense(activation="relu", units=unit, kernel_initializer="uniform"))
            i += 1
        classifier.add(Dense(units = 38, kernel_initializer = 'uniform', activation = 'softmax'))
        classifier.compile(optimizer = optimizer, loss = 'sparse_categorical_crossentropy', metrics = ['accuracy'])
        return classifier
    classifier = KerasClassifier(build_fn = build_classifier)
    parameters = {'batch_size': [10,25],
                  'epochs': [100,200],
                  'optimizer': ['adam'],
                  'nb_layers': [5,6,7],
                  'unit':[48,57,76]
                 }
    grid_search = GridSearchCV(estimator = classifier,
                               param_grid = parameters,
                               scoring = 'accuracy',
                              cv=5,n_jobs=-1)
    grid_search = grid_search.fit(X_train, y_train)
    best_parameters = grid_search.best_params_
    best_accuracy = grid_search.best_score_

推荐答案

看一下y_train的形状.它必须是形状为(，37)的某种热点

Take a look at the shape of your y_train. It need to be a some sort of one hot with shape (,37)

这篇关于我正在使用GridSearchCV训练Ann机器学习模型，并在gridSearchCV中陷入IndexError问题的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！