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...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: KONG TAOTAO, SU CHANFEI, LIN PENG
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
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