判斷numpy 中 infinite 和NAN的問題

在numpy中由於numpy太強大了,經常會計算出一些不合法的結果,若不注意放入model中訓練。就會報錯

今天教大家一個測試數據是否合法的函數

np.isnan(y)

上面的y是一個數 ,return 返回值爲True False

np.isnan(vec).any()
np.isnan(vec).all()

上面的vec是一個np array,可以判斷數組中的任意元素是否合法,特別適合放在train或者shuffle之前

np.isfinite(vec).all()

判斷是否爲無限大,vec是一個np array,return 返回值依然是 True False

np.isfinite(number)

同樣可以判斷一個數字

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章