Model training method and device, computer equipment and storage medium
The invention discloses a model training method and device, computer equipment and a storage medium, and relates to the technical field of artificial intelligence. The method comprises the steps of obtaining a training sample set; inputting each piece of sample user feature data into a first initial...
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, computer equipment and a storage medium, and relates to the technical field of artificial intelligence. The method comprises the steps of obtaining a training sample set; inputting each piece of sample user feature data into a first initial model to obtain a prediction label corresponding to each behavior prediction task; determining a first loss value corresponding to each behavior prediction task according to a difference degree between a prediction label corresponding to each behavior prediction task and a target label corresponding to each behavior prediction task; determining a total loss value based on the first loss value corresponding to each behavior prediction task; and according to the total loss value, carrying out iterative training on the first initial model until a first target condition is met, and obtaining the trained first initial model as a target marketing model. Thus, multi-task learning is adopted, the prediction capacity of th |
---|