wooner49 / T-spear-shield

Spear and Shield: Adversarial Attacks and Defense Methods for Model-Based Link Prediction on Continuous-Time Dynamic Graphs (AAAI'24)
https://arxiv.org/abs/2308.10779
3 stars 1 forks source link

Missing configuration file #2

Open ansh997 opened 3 months ago

ansh997 commented 3 months ago

Hi, Great work!! I am trying to run your code. There is a missing conf file (see below code). Can you please provide full setup to run the code?

def parse_config(model):
    config_path = f'./config/{model}.yml'
    conf = yaml.safe_load(open(config_path, 'r'))
    sample_param = conf['sampling'][0]
    memory_param = conf['memory'][0]
    gnn_param = conf['gnn'][0]
    train_param = conf['train'][0]
    return sample_param, memory_param, gnn_param, train_param
wooner49 commented 3 months ago

Thank you for your interest. We will update the config files as soon as possible.

ansh997 commented 3 months ago

Hi, gentle reminder on this. Thanks!!