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，因為只有一維。 祝你好運！