There is provided a system and method of data prediction. The method includes obtaining a hierarchical data structure comprising a plurality of layers, each including one or more nodes; obtaining one or more machine learning (ML) models each corresponding to a respective node of at least some of the nodes in at least a given layer, in response to a user's request of prediction related to a given node in the given layer; generating a prediction result using a given ML model corresponding to the given node; upon receiving the user's feedback, selecting one or more configuration parameters of the given ML model related to the feedback; updating the selected configuration parameters according to additional factors in the feedback, and re-training the given ML model to obtain a re-trained ML model,—and using the re-trained ML model to generate an updated prediction result to be sent to the user.