AttributeError:导入BERT

时间:2020-03-24 17:13:09

标签: python tensorflow bert-toolkit

仅在导入我的软件包时出现此错误。我无法找到正确的解决方案来解决此问题。任何帮助是极大的赞赏。

据我所知,这可能是Tensorflow问题?

from sklearn.model_selection import train_test_split
import pandas as pd
import tensorflow as tf
import tensorflow_hub as hub
from datetime import datetime

import bert
from bert import run_classifier
from bert import optimization
from bert import tokenization

2 个答案:

答案 0 :(得分:1)

找到这个

背景 Colab 预装了两个版本的 TensorFlow:2.x 版本和 1.x 版本。 Colab 默认使用 TensorFlow 2.x,不过您可以通过如下所示的方法切换到 1.x。

指定 TensorFlow 版本 运行 import tensorflow 将导入默认版本(当前为 2.x)。在运行 import tensorflow 之前,您可以通过使用 tensorflow_version 魔法运行单元格来使用 1.x。

[ ] %tensorflow_version 1.x 已选择 TensorFlow 1.x。

更多细节: https://colab.research.google.com/notebooks/tensorflow_version.ipynb#scrollTo=NeWVBhf1VxlH

答案 1 :(得分:0)

TensorFlow 2.x模块使用bert时,您似乎拥有TensorFlow 1.x。您可以验证here是否Tensorflow 1.x具有tf.train.Optimizer模块,而根据thisTensorflow 2.x没有这样的模块。

请确保您安装了Tensorflow所需的bert版本