AxisError: ось 1 выходит за границы массива измерения 1

Question

введите описание изображения здесь

Это мой код:

config = Config(mode='conv')

    if config.mode == 'conv':
        X, y = build_rand_feat()
        y_flat = np.argmax(y, axis=1)
        model=get_conv_model()

    elif config.mode == 'time':
        X, y = build_rand_feat()
        y_flat = np.argmax(y,axis=1)
        input_shape = (X.shape[1], X.shape[2])
        model = get_recurrent_model()

Пожалуйста, помогите мне исправить это.

Answer 1

просто изменить на ось 0, так как имеет только одно измерение. Удачи!