International Business Machines Corporation
AUTOMATED MACHINE LEARNING MODEL SELECTION

Abstract:

An approach to identifying architectures of machine learning models meeting a user defined constraint. The approach can receive input associated with evaluating machine learning models from a user. The approach can determine acceptable architectural templates to evaluate the machine learning models based on the input and determine a list of architectures and metrics based on a calculation of maximum neural network sizes of the acceptable architectural templates not exceeding the constraint. The approach can send the list of architectures and metrics to the user for selection.