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

Question

enter image description here

这是我的代码：

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，因为它只有一个尺寸。祝你好运！