logo
Loading...

不太能理解mist的Y_train =(numpy.arange(10) == y_train[:, None]).astype(int)意思 - Cupoy

不太能理解 Y_train =(numpy.arange(10) == y_train[:, Non...

ml100-2-d76,ml100-2

不太能理解mist的Y_train =(numpy.arange(10) == y_train[:, None]).astype(int)意思

2019/07/13 下午 05:03
機器學習共學討論版
鄭博元
觀看數:40
回答數:1
收藏數:0
ml100-2-d76
ml100-2

不太能理解 Y_train =(numpy.arange(10) == y_train[:, None]).astype(int) 這行的意思,原本y_train是label (0,1,2,3...),但為何最後要轉成 0,1?

回答列表

  • 2019/07/13 下午 05:24
    Jeffrey
    贊同數:0
    不贊同數:0
    留言數:0

    # 由於mist的輸入數據維度是(num, 28 , 28),這裡需要把後面的維度直接拼起來變成784維   

    不是轉成 [0, 1];