Text category prediction method and device, equipment, storage medium and program product

Abstract

The invention relates to a text category prediction method and device, computer equipment, a storage medium and a computer program product. The method relates to an automatic text classification technology of artificial intelligence, and comprises the following steps: performing semantic coding according to a semantic vector of each word in a short text through a coding layer of a trained text category prediction model to obtain a semantic coding vector of the short text; through a decoding layer of the text category prediction model, performing first decoding according to the semantic coding vector to obtain a first decoding hidden vector, and after obtaining a first category corresponding to the short text based on the first decoding hidden vector, continuing to perform current decoding according to a decoding hidden vector obtained by previous decoding and the category; obtaining a decoding hidden vector and a category of the current decoding until the decoding is finished; and combining the plurality of categories obtained by decoding according to the hierarchy to obtain the category path corresponding to the short text, so that the problem of consistency of father and child nodes of the multi-hierarchy category path can be improved, and the category path of the text can be accurately predicted.

Description

technical field [0001] The present application relates to the field of computer technology, in particular to a text category prediction method, device, computer equipment, storage medium and computer program product, and also to a text category prediction model processing method, device, computer equipment, storage medium and computer program products. Background technique [0002] With the gradual development of machine learning and deep learning techniques, significant progress has been made in classifying text using well-performing neural networks. Automatic text classification technology, especially short text classification, is widely used in various industries, such as commodity search, commodity intelligent recommendation, news classification, human-machine writing judgment, spam identification, user sentiment classification, and intelligent copywriting generation. [0003] In the multi-category classification of short texts, the categories usually have a hierarchica...

Images