Tensorflow 常用函数模块

记录一下因为tensorflow版本更新,存在部分模块在2.0 被移除或更改写法。尝试如何解决报错,正常使用tensorflow 1.0的函数。
调用相关包
  • 1)RNN

只需更改调用tensorflow库和包的写法,就OK啦

# 原1.0写法:
import tensorflow as tf
import tensorflow.contrib.rnn as rnn

# 现2.0需改成
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
from tensorflow.python.ops import rnn
#rnn_cell包同样
#from tensorflow.python.ops import rnn_cell
  • 2)CRF

出现以下报错:

from tensorflow.contrib.crf import crf_log_likelihood
(或者是调用此库"from tensorflow.contrib.crf import viterbi_decode"[解决方法同理])
ModuleNotFoundError: No module named 'tensorflow.contrib'

Tensorflow2.0 将tf.contrib移除,转移至第三方库

# 原1.0写法:
from tensorflow.contrib.crf import crf_log_likelihood

# 现2.0需改成
from tensorflow_addons.text.crf import crf_log_likelihood
"""
对于tensorflow_addons的下载:pip install tensorflow-addons
ps:注意:tfa版本需要安装对应Tensorflow版本,若tf为最新,则直接上面那样pip ↑ 就可以啦
"""

CRP函数理解,参考链接

函数模块
  • 1)Session函数

只需更改调用tensorflow库的写法

# 原1.0写法:
import tensorflow as tf
with tf.Session() as sess:
    print(sess.run(result))

# 现2.0需改成
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
with tf.Session() as sess:
    print(sess.run(result))
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值