Machine learning model training method, device, storage medium and computer device

Abstract

A machine learning model training method, apparatus, storage medium and compute device are disclosed. That method includes: obtaining model training data and correspond tag; The model training data includes feature data of a user sample and feature data of a recommended object sample; Determining a weight of the model training data according to a generation time of the model training data; Inputting the model training data into a machine learning model to obtain a user behavior prediction result corresponding to the user sample and the recommendation object sample; Determining a training target according to a difference between the user behavior prediction result and the tag and the weight; According to the direction of optimizing the training objective, the model parameters of the machinelearning model are adjusted and the training is continued until the training is finished when the training stopping condition is satisfied. The prediction effect of the machine learning model trainedby the scheme provided by the present application is good.

Description

technical field [0001] The present application relates to the field of computer technology, in particular to a machine learning model training method, device, storage medium and computer equipment. Background technique [0002] With the development of computer technology, in the process of recommending objects, machine learning models are usually used to make recommendation predictions for recommended objects. Before using a machine learning model to make recommendation predictions on recommended objects, it is necessary to train a machine learning model for recommendation prediction. [0003] However, in the traditional machine learning model training process, the differences between training samples are not taken into account, resulting in inaccurate predictions when using the machine learning model trained by traditional methods for recommendation prediction. Contents of the invention [0004] Based on this, it is necessary to provide a machine learning model training ...

AI Technical Summary

Problems solved by technology

[0004] Based on this, it is necessary to provide a machine learning model training method, device, storage medium and computer equipment for the problem of inaccurate predictions that may occur when using a machine learning model trained by a traditional method for recommendation prediction.

Images