Prediction model training method and device and electronic equipment

A prediction model and training method technology, applied in the field of machine learning, can solve the problems of high overhead cost, low prediction accuracy, low multi-value attribute support, etc., and achieve the effect of reducing overhead cost and simplifying deployment steps.

Pending Publication Date: 2020-07-03
BEIJING QIHOO TECH CO LTD
View PDF0 Cites 4 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

Therefore, the online processing of the existing DeepFM model has the problem of high overhead cost
In addition, the existi

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Prediction model training method and device and electronic equipment
  • Prediction model training method and device and electronic equipment
  • Prediction model training method and device and electronic equipment

Examples

Experimental program
Comparison scheme
Effect test

Example Embodiment

[0067] Example one

[0068] The embodiment of the application provides a method for training a prediction model, such as figure 1 As shown, the method includes: step S110 to step S140.

[0069] Step S110: Obtain sample data.

[0070] In this embodiment, the sample data can be obtained by custom editing, or collected from a terminal device. In actual application, sample data is generally obtained from terminal equipment. Specifically, the sample data may include user personal information corresponding to the terminal device, detailed information of applications installed on the terminal device, and so on. In actual application, the user's personal information may include age, gender, hobbies, etc.; the detailed information of the application program may characterize the application programs installed on the terminal device, such as QQ music, XX security guard, office, etc.

[0071] In practical applications, the terminal device can be an electronic device such as a PC, a notebook, a ...

Example Embodiment

[0113] Example two

[0114] An embodiment of the application provides a training device for a prediction model, such as Figure 5 As shown, the device 50 may include: a data acquisition module 501, a vector conversion module 502, a model adjustment module 503, and a model training module 504, where:

[0115] The data acquisition module 501 is used to acquire sample data;

[0116] The vector conversion module 502 is used to convert the sample data into corresponding vector data by using the pre-built DeepFM model;

[0117] The model adjustment module 503 is configured to obtain the position correspondence of any attribute feature in the sample data in the vector data, and store it in the DeepFM model to obtain the adjusted DeepFM model;

[0118] The model training module 504 is configured to train the adjusted DeepFM model by using vector data to obtain the DeepFM prediction model, so as to deploy the DeepFM prediction model.

[0119] In the embodiment of the present application, sample d...

Example Embodiment

[0137] Example three

[0138] The embodiment of the application provides an electronic device, such as Image 6 As shown, Image 6 The illustrated electronic device 600 includes a processor 6001 and a memory 6003. Among them, the processor 6001 and the memory 6003 are connected, such as by a bus 6002. Further, the electronic device 600 may also include a transceiver 6006. It should be noted that in actual applications, the transceiver 6006 is not limited to one, and the structure of the electronic device 600 does not constitute a limitation to the embodiments of the present application.

[0139] The processor 6001 may be a CPU, a general-purpose processor, DSP, ASIC, FPGA or other programmable logic devices, transistor logic devices, hardware components, or any combination thereof. It can implement or execute various exemplary logical blocks, modules and circuits described in conjunction with the disclosure of this application. The processor 6001 may also be a combination that...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

PUM

No PUM Login to view more

Abstract

The embodiment of the invention provides a prediction model training method and device and electronic equipment. The method comprises the following steps: acquiring sample data; converting the sampledata into corresponding vector data by using a pre-constructed DeepFM model; obtaining a position corresponding relationship of any attribute feature in the sample data in the vector data, and storingthe position corresponding relationship in theDeepFM model to obtain an adjusted DeepFM model; and training the adjusted DeepFM model by using the vector data to obtain a DeepFM prediction model so as to deploy the DeepFM prediction model. According to the embodiment of the invention, the method solves a problem in the prior art that the operation is complex because an index file needs to be built and the format conversion needs to be carried out in an online DeepFM model deployment process, simplifies the deployment steps, and reduces the cost of the online application of the DeepFM prediction model.

Description

technical field [0001] The present application relates to the technical field of machine learning, and in particular, the present application relates to a training method, device and electronic equipment for a predictive model. Background technique [0002] With the development of machine learning algorithms, personalized recommendation functions are widely used. The effect of machine learning algorithms is greatly affected by features, and the work of feature engineering requires a lot of human and material costs, especially the extraction and screening of combined features. In the existing technology, DeepFM is generally used to complete feature extraction and screening. The working principle of DeepFM is: combine deep learning and factorization machine algorithm, and automatically combine first-order, second-order and high-order features, thereby reducing feature engineering. cost and improve the algorithm effect. [0003] The existing DeepFM model has format requiremen...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

Application Information

Patent Timeline
no application Login to view more
IPC IPC(8): G06F16/903G06K9/62G06N3/08
CPCG06N3/08G06F18/214
Inventor 翟羽行
Owner BEIJING QIHOO TECH CO LTD
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Try Eureka
PatSnap group products