from tensorflow.contrib import learn换成import tf_slim as slim后,learn.preprocessing.VocabularyProcessor报错
时间: 2024-04-30 08:18:52 浏览: 189
Tensorflow 利用tf.contrib.learn建立输入函数的方法
`tf_slim` 和 `tensorflow.contrib` 有些不同,所以需要针对性地修改代码。
`tf_slim` 中没有 `learn.preprocessing.VocabularyProcessor` 这个类,但是我们可以使用 `tf.contrib.learn.preprocessing.VocabularyProcessor` 替代。
你可以将代码修改为以下形式:
```python
from tensorflow.contrib import learn
import tensorflow as tf
import tf_slim as slim
# 使用 tf.contrib.learn.preprocessing.VocabularyProcessor
vocab_processor = learn.preprocessing.VocabularyProcessor(max_document_length)
x = np.array(list(vocab_processor.fit_transform(x_text)))
# 其他代码不变
```
另外,你也可以使用 `tf.data.Dataset` API 来处理数据,这是 TensorFlow 推荐的数据输入方式,也更加方便和高效。
阅读全文