Method and device for training interaction prediction model and method and device for predicting interaction object

Abstract

The embodiment of the invention provides an method and device for training an interaction prediction model and a method and device for using the interaction prediction model. The training method comprises the steps: firstly, constructing a dynamic interaction graph based on an interaction event sequence,determining a first node and a second node corresponding to a current interaction event from the dynamic interaction graph, wherein the first node points to two historical nodes corresponding to a previous interaction event in which the first node participates; using a graph coding network to obtain coding vectors corresponding to the two historical nodes; and inputting the coding vectors and the respective occurrence moments of the two interaction events into a prediction representation network to obtain a first representation vector of the first node; inputting the attribute characteristics of the second node into an attribute coding network to obtain a second attribute vector; predicting the interaction probability according to the first representation vector and the second attribute vector, determining the loss according to the interaction probability, and updating the graph coding network, the prediction representation network and the attribute coding network.

Description

technical field [0001] One or more embodiments of this specification relate to the field of machine learning, and in particular to training an interaction prediction model, and a method and device for predicting interactive objects using the trained interaction prediction model. Background technique [0002] In many scenarios, user interaction events need to be analyzed and processed. Interaction events are one of the basic elements of Internet events. For example, the click behavior when a user browses a page can be regarded as an interaction event between the user and the content block of the page, and the purchase behavior in e-commerce can be regarded as the interaction event between the user and the product. The interaction event between accounts, and the transfer behavior between accounts is the interaction event between users. A series of user interaction events contains the characteristics of the user's fine-grained habits and preferences, as well as the characteris...

AI Technical Summary

Problems solved by technology

[0003] However, the interaction event involves both parties, and the state of each participant can change dynamically. Therefore, it is very difficult to accurately express the characteristics of the interaction participants by comprehensively considering the various characteristics of the interaction participants.

Images