Learning Weighted Model Integration Distributions