AI-based modelling tool