Add weight distribute by machine learing