albert-bilstm-crf命名实体识别

Question

最近在做命名实体识别，复现albert-bilstm-crf模型，从github上下载了代码，环境应该是配置好了。运行出现下面的问题，请大家帮忙解答一下

Using TensorFlow backend.
2024-03-03 09:45:24.812480: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
{'O': 1, 'B-LOC': 2, 'I-LOC': 3, 'B-PER': 4, 'I-PER': 5, 'B-ORG': 6, 'I-ORG': 7} 128 D:/notebook/albert/ALBERT_NER_KERAS-master/data/example.train D:/notebook/albert/ALBERT_NER_KERAS-master/data/example.dev D:/notebook/albert/ALBERT_NER_KERAS-master/data/example.test {1: 'O', 2: 'B-LOC', 3: 'I-LOC', 4: 'B-PER', 5: 'I-PER', 6: 'B-ORG', 7: 'I-ORG'}
E:BERT_VEC:[graph:opt:149]:fail to optimize the graph!
E:BERT_VEC:[graph:opt:150]:module 'tensorflow' has no attribute 'ConfigProto'
Exception in thread Thread-1:
Traceback (most recent call last):
  File "D:\Anaconda3\envs\test\lib\threading.py", line 916, in _bootstrap_inner
    self.run()
  File "D:\Anaconda3\envs\test\lib\threading.py", line 864, in run
    self._target(*self._args, **self._kwargs)
  File "D:\notebook\albert\ALBERT_NER_KERAS-master\albert_zh\extract_feature.py", line 92, in predict_from_queue
    for i in prediction:
  File "D:\Anaconda3\envs\test\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 626, in predict
    features, None, ModeKeys.PREDICT, self.config)
  File "D:\Anaconda3\envs\test\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 1152, in _call_model_fn
    model_fn_results = self._model_fn(features=features, **kwargs)
  File "D:\notebook\albert\ALBERT_NER_KERAS-master\albert_zh\extract_feature.py", line 69, in model_fn
    graph_def.ParseFromString(f.read())
  File "D:\Anaconda3\envs\test\lib\site-packages\tensorflow_core\python\lib\io\file_io.py", line 122, in read
    self._preread_check()
  File "D:\Anaconda3\envs\test\lib\site-packages\tensorflow_core\python\lib\io\file_io.py", line 84, in _preread_check
    compat.as_bytes(self.__name), 1024 * 512)
  File "D:\Anaconda3\envs\test\lib\site-packages\tensorflow_core\python\util\compat.py", line 87, in as_bytes
    (bytes_or_text,))
TypeError: Expected binary or unicode string, got None

以下是我的主要环境依赖库版本：
joblib 1.1.1
Keras 2.3.1
Keras-Applications 1.0.8
keras-contrib 2.0.8
Keras-Preprocessing 1.1.2
numpy 1.19.5
pandas 1.1.5
tensorflow 2.1.0
tensorflow-estimator 2.1.0
cuda 10.1
cudnn 8.0.5

Answer 1

分毫析厘 上午好☀️☀️☀️️
本答案参考ChatGPT-3.5

根据错误提示和环境依赖库版本，可以初步判断问题出在tensorflow和tensorflow_estimator版本之间的不兼容性上。解决方案如下：

1. 根据错误提示 "module 'tensorflow' has no attribute 'ConfigProto'"，可以查看tensorflow版本是否为2.1.0，如果不是，则需要将tensorflow版本切换为2.1.0。

2. 在代码中存在使用tensorflow_estimator库，该库版本需要与tensorflow版本保持一致。可以尝试将tensorflow_estimator版本更改为2.1.0。

3. 检查tensorflow和tensorflow_estimator的版本兼容性是否满足，可以在tensorflow官方文档中查找对应版本之间的兼容关系。

4. 如果以上两个版本已经满足兼容关系，可以尝试卸载tensorflow和tensorflow_estimator库，然后重新安装。

5. 检查依赖库的其他版本是否与tensorflow和tensorflow_estimator兼容。

6. 检查环境变量是否设置正确。

注意：以上解决方案都基于初步判断，如果问题还未解决，可以进一步调查其他可能的原因，如代码中是否存在其他不兼容的库、代码本身的问题等等。

修改后的代码示例：

import tensorflow as tf
...

config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)
...

希望这些解决方案可以帮助您解决问题。如果问题还未解决，请提供更多的信息，以便我们进一步帮助您解决。