The invention discloses a
traditional Chinese medicine recommendation method based on knowledge driving and a residual
attention network, and the method comprises the following steps: S1, a data collection and cleaning module: obtaining prescription text data, executing a standardized process for symptoms and
traditional Chinese medicine terms in the prescription text data, and dividing a
data set after the standardized process is executed into a
training set, a
verification set and a
test set; s2, an entity pre-training module: extracting symptom semantic features by using a Word2vec training model to extract the features of symptom entities in the prescription and obtain overall context
semantic representation, constructing a
traditional Chinese medicine attribute
knowledge graph, and fusing the natural attribute features of the traditional Chinese
medicine as external knowledge into a
recommendation model; s3, a multi-graph construction and entity
feature learning module: according to a co-occurrence relationship between the symptoms and the traditional Chinese medicines, constructing a symptom-traditional Chinese
medicine isomerism graph (SHHG), a symptom-symptom isomerism graph (SSIG) and a traditional Chinese
medicine-traditional Chinese medicine isomerism graph (HHIG); respectively extracting symptom features, traditional Chinese medicine features and interaction features from a symptom-traditional Chinese medicine isomeric graph (SHHG), a symptom-symptom isocomposition (SSIG) and a traditional Chinese medicine-traditional Chinese medicine isocomposition (HHIG) by using a graph attention neural network, and enhancing entity features by using a residual structure; the
feature fusion and traditional Chinese medicine recommendation module is used for fusing symptom features and traditional Chinese medicine features extracted from a symptom-traditional Chinese medicine isomerism graph (SHHG), a symptom-symptom isomerism graph (SSIG) and a traditional Chinese medicine-traditional Chinese medicine isomerism graph (HHIG) to obtain final symptom features and traditional Chinese medicine features, and expressing the symptom features by using MLP to obtain a traditional Chinese medicine recommendation result. And finally recommending a proper traditional Chinese medicine set according to the given symptom set.