A method and apparatus for training a model

Abstract

A method and apparatus for training a model is disclosed in embodiment of the present application. One embodiment of the method includes obtaining a configuration file, wherein the configuration filecomprises a data set identification, a machine learning framework identification, a machine learning algorithm identification and parameter information; selecting data from the data set indicated by the data set identification as training data to generate a training data set; selecting the machine learning framework indicated by machine learning framework identification as the target machine learning framework from the preset set of machine learning frameworks; in the framework of target machine learning, using the machine learning algorithm to identify the indicated machine learning algorithmand the parameters indicated by the parameter information, and obtaining the classification model by training based on the training data set and the labels associated with the training data in the training data set. The embodiment enables users to select different machine learning frameworks, corresponding machine learning algorithms and parameters according to actual needs, and expands the application scope of model training.

Description

technical field [0001] The embodiments of the present application relate to the field of computer technologies, and in particular to methods and devices for training models. Background technique [0002] As machine learning is gradually adopted by various industries, general-purpose machine learning models often cannot meet users' customized AI (Artificial Intelligence, artificial intelligence) needs. Users usually think of using specific data to customize and train their own models to suit specific scenarios, such as image classification, object detection, voice recognition, voiceprint recognition, and so on. Since different types of models may need to be trained under different types of machine learning frameworks, and the existing model training systems usually support limited machine learning frameworks, the existing model training systems generally cannot meet the different model training needs of users . Contents of the invention [0003] Embodiments of the present...

AI Technical Summary

Problems solved by technology

Since different types of models may need to be trained under different types of machine learning frameworks, and the existing model training systems usually support limited machine learning frameworks, the existing model training systems generally cannot meet the different model training needs of users

Images