Ancient poem generation method and device based on constraint decoding, equipment and medium

Abstract

The invention relates to the field of artificial intelligence, and provides an ancient poem generation method and device based on constraint decoding, and equipment and a medium, and the method comprises the steps of obtaining a poem type and an original text in an ancient poem generation instruction; carrying out word segmentation and sensitive word removal processing on the original text to obtain segmented words, and obtaining a TF-IDF value of each segmented word through a TF-IDF algorithm; inputting all the segmented words and the TF-IDF values corresponding to the segmented words into akeyword recognition model, and performing recognition extension processing through the keyword recognition model to obtain keywords; sorting all the keywords through a TextRank algorithm, and determining a final keyword; and inputting the poem type and all the final keywords into an ancient poem generation model based on Beam search and seq2seq, and generating an ancient poem result correspondingto the poem type through encoding and constrained decoding processing. According to the invention, the ancient poem conforming to the poem and the poem type of the original text is automatically generated, the accuracy and reliability of generating the poem are improved, and the quality of generating the ancient poem is improved.

Description

technical field [0001] The present invention relates to the field of natural language processing of artificial intelligence, in particular to a method, device, equipment and medium for generating ancient poems based on constraint decoding. Background technique [0002] In the existing ancient poetry generation scheme, a text content is generally input into a deep learning-based poetry model, and the poem is generated through the poem model. The disadvantage of this scheme is that repeated words, Problems such as non-compliance with the meter and rhyme requirements, etc., lead to the lack of poetic meaning of the generated verses, and there are various problems such as incoherence, resulting in the generated verses not conforming to the poetic meaning of the user, and the generation accuracy is low, which in turn affects the user's satisfaction degree, and lead to poor user experience and other phenomena. Contents of the invention [0003] The present invention provides an...

AI Technical Summary

Problems solved by technology

[0002] In the existing ancient poetry generation scheme, a text content is generally input into a deep learning-based poetry model, and the poem is generated through the poem model. The disadvantage of this scheme is that repeated words, Problems such as non-compliance with the meter and rhyme requirements, etc., lead to the lack of poetic meaning of the generated verses, and there are various problems such as incoherence, resulting in the generated verses not conforming to the poetic meaning of the user, and the generation accuracy is low, which in turn affects the user's satisfaction degree, and lead to poor user experience and other phenomena

Images