Machine learning enhanced resource allocation in wireless networks