Artificial intelligence based traffic management