reweight-a-probability-distribution.py

import numpy as np


def reweight_distribution(original_distribution, temperature=0.5):
    distribution = np.log(original_distribution) / temperature
    distribution = np.exp(distribution)

    return distribution / np.sum(distribution)