def policy_fn(name, ob_space, ac_space, disease_dim):
return gcn_policy.GCNPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
disease_dim=disease_dim,
atom_type_num=env1.atom_type_num,
args=args)
def policy_fn(name, ob_space, ac_space):
return gcn_policy.GCNPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
atom_type_num=env.atom_type_num,
char_type_num=len(env.smile_chars),
args=args)
def policy_fn(name, ob_space, ac_space):
return gcn_policy.GCNPolicy(name=name,
ob_space=ob_space,
ac_space=ac_space,
atom_type_num=env.atom_type_num,
args=args)
