Modeling hydrodynamic interactions in non-spherical particle-laden flows : numerical simulations and machine learning models