行秩列秩矩阵的秩 2021-03-01 19:29 采纳率: 50%
浏览 651
已采纳

使用kashgari实现BERT+Bilstm命名实体识别,在保存模型时报错!!!求助!!!

 以下是我的代码,

import tensorflow as tf
import time
import jieba as jb
import random
import kashgari
import sys,io
from kashgari.corpus import ChineseDailyNerCorpus
from kashgari.tasks.labeling import BiLSTM_Model
from kashgari.embeddings import BertEmbedding

sys.stdout = io.TextIOWrapper(sys.stdout.buffer,encoding='utf-8') # Change default encoding to utf8

start = time.process_time()
train_x,train_y = ChineseDailyNerCorpus.load_data('train')
test_x, test_y = ChineseDailyNerCorpus.load_data('test')
valid_x, valid_y = ChineseDailyNerCorpus.load_data('valid')

embedding = BertEmbedding('chinese_L-12_H-768_A-12')
model = BiLSTM_Model(embedding,sequence_length=100)
model.fit(train_x,train_y,valid_x,valid_y,epochs=1)

model.save('model_learn2/bilstm_ner')

end = time.process_time()
step = end - start
print("总共耗时:%0.3f 秒,相当于 %0.3f 分钟" % (step,step / 60))

结果报出这样的错误

我的TensorFlow版本是2.1.0;kashgari版本是2.0.1;BERT, Chinese 中文模型使用的是Google Cloud的BERT-base, Chinese

  • 写回答

5条回答 默认 最新

  • ProfSnail 2021-03-01 21:04
    关注

    你好。这个错误需要进入源码进行一下修正。我已经私信你了,请看一下。

    需要在D:\dev\anaconda\lib\site-packages\kashgari\tasks\abs_task_model.py的82行open(filename)as f;的时候修改为open(filename, encoding='utf-8') as f. 

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(4条)

报告相同问题?

悬赏问题

  • ¥15 MddBootstrapInitialize2失败
  • ¥15 LCD Flicker
  • ¥15 Spring MVC项目,访问不到相应的控制器方法
  • ¥15 esp32在micropython环境下使用ssl/tls连接mqtt服务器出现以下报错Connected on 192.168.154.223发生意外错误: 5无法连接到 MQTT 代理,如何解决?
  • ¥15 关于#genesiscsheel#的问题,如何解决?
  • ¥15 Android aidl for hal
  • ¥15 STM32CubeIDE下载程序报错
  • ¥15 微信好友如何转变为会员系统?(相关搜索:小程序)
  • ¥15 c# 直接使用c++ 类库文件
  • ¥15 一个主机电脑有两个显示器,当前有两个软件主体是网页html打包的exe程序,如何通过cmd命令bat同时打开软件一个在主屏幕显示,另外一个软件在2副屏幕上显示