版本:

OS : Ubuntu 14.04 LTS

tensorflow 1.2

keras 2.2.1

在調用softmax函數時出現 TypeError: softmax() got an unexpected keyword argument 'axis' 的錯誤,其主要原因是因為keras在將語法轉換成tensorflow時某些部分沒有對應上,主因應該是版本落差導致,若不透過更新到匹配版本的情況下,可以修改keras 套件中的部分程式碼,使其匹配.

 

而從錯誤訊息中可以發現,問題檔案為:

/usr/local/lib/python2.7/dist-packages/keras/backend

 

因此使用編輯器(vim/nano) 編輯 tensorflow_backend.py

找到3458行中的 axis=axis.

並將其移除,修改成果如下圖:

修改完畢後正常情況即可使用.

 

 

arrow
arrow
    文章標籤
    keras
    全站熱搜

    Lung-Yu,Tsai 發表在 痞客邦 留言(0) 人氣()

    E-mail轉寄