AxisError：轴 1 超出维度 1 数组的范围

Question

enter image description here

This is my code:

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()

Please help me to fix this.

Answer 1

simply change to axis 0, as has only one dimension. Good Luck!