关于tensorflow v1和v2版本遇到的问题

最新推荐文章于 2024-03-04 23:43:06 发布

seven_不是赛文

最新推荐文章于 2024-03-04 23:43:06 发布

阅读量580

点赞数

分类专栏： python实操文章标签： tensorflow lstm 深度学习

本文链接：https://blog.csdn.net/weixin_51015047/article/details/122856473

版权

python实操专栏收录该内容

62 篇文章 3 订阅

订阅专栏

AttributeError: module tensorflowhas no attribute placeholder

报错原因：tensorflow 2.0版本去掉了placeholder，而tensorflow 1.*版本才有。

import tensorflow as tf
改成：
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

这类语句：
X=tf.placeholder(“float”) Y=tf.placeholder(“float”)

改为：
X=tf.compat.v1.placeholder(“float”)
Y=tf.compat.v1.placeholder(“float”)

module ‘tensorflow.compat.v1’ has no attribute ‘contrib’

lstm = tf.contrib.rnn.BasicLSTMCell(lstm_size)
drop = tf.contrib.rnn.DropoutWrapper(lstm, output_keep_prob=keep_prob_) 
cell = tf.contrib.rnn.MultiRNNCell([drop] * lstm_layers)

解决办法：
打开tensorflow官方网站，在搜索框里边分别搜索BasicLSTMCell,DropoutWrapper,MultiRNNCell，
在这里插入图片描述

可以看到需要使用tf.compat.v1.nn.rnn_cell.MultiRNNCell来替代,其他同理。将以上代码修改为

lstm = tf.nn.rnn_cell.BasicLSTMCell(lstm_size)
drop = tf.nn.rnn_cell.DropoutWrapper(lstm, output_keep_prob=keep_prob_)
cell = tf.nn.rnn_cell.MultiRNNCell([drop] * lstm_layers)