Meta Platforms, Inc.
SYSTEM AND METHODS FOR FEATURE ENGINEERING BASED ON GRAPH LEARNING

Abstract:

In one embodiment, a computing system may receive query information associated with a machine-learning model. The system may access a knowledge graph that defines relationships between a number of machine-learning models and a number of features of the machine-learning models. The system may determine, based on the knowledge graph and the query information, one or more correlation metrics indicating correlations between the machine-learning model and one or more features of the features in the knowledge graph. The system may determine one or more recommended features for the machine-learning model based on the one or more correlation metrics and the one or more features.