User consumption behavior prediction model training method and device, equipment and storage medium

Abstract

The invention discloses a user consumption behavior prediction model training method and device, equipment and a storage medium, and relates to the technical field of machine learning model training.The method includes: obtaining training set and a test set from a database; obtaining a prediction model initialization weight vector, an inverse connection function and a learning rate parameter; forany training data, normalizing the weight vector and constructing a first random variable, sampling the first random variable for multiple times to obtain a first inner product estimated value, and updating the weight vector according to the inverse connection function, the first inner product estimated value, the label information of the training data and the learning rate parameter; and testingeach weight vector by adopting test data, and obtaining a prediction model of the trained user consumption behavior according to the weight vector with the minimum risk of the prediction model. In the generalized linear model training process, inner product estimation is approximated by sampling random variables multiple times, the model training efficiency is improved, the model accuracy is ensured, and the generalized linear model can effectively predict user consumption behaviors.

Description

technical field [0001] The present application relates to the field of computer technology, in particular to the technical field of machine learning model training. Background technique [0002] Generalized linear model (GLM) is a flexible linear regression model, which is a very basic and widely used method in machine learning. The generalized linear model establishes the relationship between the mathematical expectation value of the random variable measured by the experimenter and the predictor variable of the linear combination through the link function. Its model assumes that the output y and each input vector x have a linear relationship after acting on the link function, that is, y=g(w x), where g is the inverse link function (the inverse of the link function), w x is the inner product of the weight vector w and the input vector x. The core of the technical problem of generalized linear model learning is to design an efficient scheme to learn the weight vector w from...

AI Technical Summary

Problems solved by technology

[0004] The existing sparse sub-method runs slowly, seriously affecting the training efficiency of the generalized linear model, especially when the generalized linear model is used to predict user consumption behavior, due to the high dimensionality of the input data, the training speed is slow and the training process takes a long time; However, the sample complexity of the GLMtron method is not optimal. To achieve the same accuracy, more samples are required than the sparse sub-method, that is, a larger training set.

Images