Model training method and device and computer readable storage medium

The invention discloses a model training method and device and a computer readable storage medium, and the method comprises the steps: obtaining graph data, wherein the graph data comprises a plurality of graph nodes, a node label of each graph node, and an edge for connecting the graph nodes; perfo...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: DONG JINGRAN, QI YIJIE, ZHANG JIE, HUANG WEN, CHEN SHOUZHI, CHEN CHUAN
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:The invention discloses a model training method and device and a computer readable storage medium, and the method comprises the steps: obtaining graph data, wherein the graph data comprises a plurality of graph nodes, a node label of each graph node, and an edge for connecting the graph nodes; performing feature extraction on each graph node by adopting a graph neural network model to obtain a node feature of each graph node; classifying the plurality of graph nodes according to the node label of each graph node, and calculating a class center feature corresponding to each graph node based on a classification result and the node feature of each graph node; constructing a positive sample pair and a negative sample pair of comparative learning based on the node feature of each graph node and the class center feature corresponding to each graph node; and training the graph neural network model according to training samples constructed by the positive sample pair and the negative sample pair to obtain a trained g