跑keras模型涉及到的 keras降级

       今天在服务器上跑一个HAN(Hierachical Attention network) 分类模型,使用的keras库,然后报错: TypeError: softmax() got an unexpected keyword argument 'axis'.   

      这种一般就是库版本的问题,升级了keras,tensorflow之后就可能有问题.  发现 tensorflow 和 keras 版本升级之后,api各种变.函数所在的包会变,函数名会变,函数的参数的顺序也要变一下,完全不能兼容老版本.有点坑.

       google了一下,有人说可以把 tensorflow降级到 1.4.0,keras 降级到 2.0.8.

pip uninstall keras
pip install  keras==2.0.8 安装指定版本 
#pip install --upgrade keras==2.0.8 升级到指定版本

       keras降级之后模型可以正常跑了.