In this extended abstract, we propose an Artificial Intelligence-based model dedicated to the representation of a multi-class traffic flow, i.e. a traffic flow in which different vehicle classes (at least cars and trucks) are explicitly represented, with the aim of using it for the development of freeway traffic control schemes based on ramp management. Specifically, the goal of this work is to develop a hybrid modelling technique in which a Machine Learning component and the multi-class version of METANET model are adopted to determine a better estimation and forecasting tool for freeway traffic. The resulting model is specifically devised in order to be included in a Model Predictive Control (MPC) scheme for the required traffic state prediction.