def build_tasker(args, dataset): if args.task == 'link_pred': return lpt.Link_Pred_Tasker(args, dataset) elif args.task == 'edge_cls': return ect.Edge_Cls_Tasker(args, dataset) elif args.task == 'node_cls': return nct.Node_Cls_Tasker(args, dataset) elif args.task == 'static_node_cls': return nct.Static_Node_Cls_Tasker(args, dataset) else: raise NotImplementedError('still need to implement the other tasks')
def build_tasker(args,dataset): if args.task == 'link_pred': return lpt.Link_Pred_Tasker(args,dataset) else: raise NotImplementedError('still need to implement the other tasks')