Machine Learning Models for Enhancing Flow in Demand-Driven Networks