基于BERT的海上船舶安全隐患分类

为准确分析出影响海上船舶安全的隐患类别，提高船舶、船员、设备的管理制度，提出一种基于双向编码器（BERT）的海上船舶安全隐患分类。采用Text Rank算法对隐患文本进行关键词提取，将提取的关键词输入到BERT预训练模型，提升模型在分类任务上的性能。试验结果表明，训练前采用Text Rank算法进行关键词提取，准确率、召回率和F1值都有所提升，验证了模型的有效性，达到了快速准确对海上船舶安全隐患分类的目的。