Как исправить ошибку <module>для моей нейронной сети?

Question

Я получаю следующую ошибку Файл "C: /Users/xx/Desktop/xxx/ost.py", строка 213, в main (), при попытке инициировать обучение нейронной сети

Это для окончательного лабораторного задания о классификации изображений

if __name__ == '__main__':
     main() 

и основная функция

def main():
    with open('train.pkl', 'rb') as f:
        data = pickle.load(f)

    x = data[0]
    y = data[1]

    # x=feature_extraction_hog(x)

    D = x.shape[1]

    hidden_size = 55
    output_size = 36

    w1 = np.random.uniform(low=-0.1, high=0.1, size=(hidden_size, (D + 1)))
    w2 = np.random.uniform(low=-0.1, high=0.1, size=(output_size, (hidden_size + 1)))

    w1, w2 = fit(x, y, True, output_size, hidden_size, w1, w2, 0.2, 0.2)

    np.savetxt('weights1.txt', w1)
    np.savetxt('weights2.txt', w2)

    y_train_pred = predict(x, w1, w2)

    print(classification_error(y_train_pred, y))

Answer 1

Мне удалось это исправить, проблема была в строке, которая получила слишком много индексов!В любом случае, спасибо