Machine learning model repository management and search engine

Abstract

Mechanisms are provided to implement a machine learning framework that operates to register a plurality of machine learning algorithms used to train machine learning models to perform related tasks, and to index the machine learning algorithms to generate and store a machine learning algorithm metadata model for each machine learning algorithm. The machine learning framework receives a user specification of an analytics pipeline task for which a machine learning model is to be trained, and converts the user specification to machine learning algorithm search criteria used to search the index to identify matching machine learning algorithms having a corresponding machine learning algorithm metadata model that matches the machine learning algorithm search criteria. The machine learning framework outputs information describing the matching machine learning algorithms.