Esempi in Python per Link_Pred_Tasker

Esempio n. 1

File: run_exp.py Progetto: junwei-h/EvolveGCN

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')

Esempio n. 2

File: run_exp.py Progetto: sykailak/evolvegcn

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')