A system described herein may provide a technique for the augmentation of one or more models based on target distributions of values associated with the models. Such models may be associated with a wireless network, and may associate a set of states to a set of values and/or to a set of actions. Such states may include particular sets of configuration parameters associated with the wireless network, the values may include Key Performance Indicators (“KPIs”) associated with a randomness state model or other metrics associated with the wireless network, and the actions may include one or more actions to perform with respect to the wireless network when given a particular state and/or set of KPIs.