change the uniform distribution to learnt distribution

Signed-off-by: Enliang Zheng <ezheng@cs.unc.edu>