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...
Gespeichert in:
Hauptverfasser: | , , , , , |
---|---|
Format: | Patent |
Sprache: | chi ; eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
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 |
---|