Poem generation method and device, computer equipment and storage medium

Abstract

The invention relates to artificial intelligence, and provides a poem generation method comprising the following steps: obtaining theme information of a poem to be generated; inputting the theme information into a poem prediction model to generate an initial candidate poem set; inputting candidate poems in the initial candidate poem set into a preset n-gram language model to generate a target candidate poem set; carrying out escort screening processing on candidate poems of the target candidate poem set according to a preset escort rule, and calculating a repetition rate of each poem to be screened; and carrying out poem repetition rate screening on the plurality of poems to be screened according to the repetition rate to obtain a target poem. Through adoption of the poem search method, the initial candidate poem set obtained by predicting the poem search method is screened through the poem fluency, the poem escort and the poem repetition rate, the smoothness, the escort performance and the non-repeatability of the generated poems are improved, and the problem of poor poem generation quality in the prior art is solved.

Description

technical field [0001] The present invention relates to the field of artificial intelligence, in particular to a poetry generation method, device, computer equipment and storage medium. Background technique [0002] Poetry is a part of the essence of Chinese culture. After a long period of accumulation, many high-quality ancient poems and modern poems have been produced. With the rapid development of deep learning technology, models based on deep learning have also been applied to poetry creation. The poetry prediction model realizes the function of automatically generating poetry by learning the existing ancient or modern poetry corpus, learning their writing characteristics and methods, and then imitating, creating, and regenerating. [0003] At present, the poetry generation model based on deep learning uses a sentence-by-sentence cycle to generate poems. Specifically, first construct a training set for poetry generation through keywords, and then predict the current lin...

AI Technical Summary

Problems solved by technology

[0004] The present invention provides a poetry generation method, device, computer equipment and storage medium to solve the problem of poor quality of generated poetry

Images