Predictive machine learning feature selection

Abstract

Various embodiments of the present disclosure provide feature engineering techniques for improving machine learning model development, performance, and maintenance. The feature engineering techniques include generating a model description vector from a textual model description for a target machine learning model and using the model description vector to generate one or more description-based similarity vectors that each include one or more of a plurality of similarity scores for a plurality of machine learning features. The feature engineering techniques include generating a label-based similarity vector based on a comparison between training data for the target machine learning model and a plurality of feature values of the plurality of machine learning features. The feature engineering techniques include providing a predictive feature set for the target machine learning model based on the one or more similarity vectors and the one or more label-based similarity vectors.