def get_reward_funcs(forward_path, backward_path, *args, **kwargs):
net_forward = Seq2Seq.load(forward_path, *args, **kwargs)
net_backward = Seq2Seq.load(backward_path, *args, **kwargs)
minfo = MutualInformation(net_forward, net_backward)
eanswer = EaseAnswering(net_forward)
iflow = InformationFlow()
return minfo, eanswer, iflow
def get_mutual_information(forward_path, backward_path, *args, **kwargs):
net_forward = Seq2Seq.load(forward_path, *args, **kwargs)
net_backward = Seq2Seq.load(backward_path, *args, **kwargs)
return MutualInformation(net_forward, net_backward)