SYMBOL INDEX (992 symbols across 175 files)

FILE: examples/linkproppred/tgbl-coin/dyrep.py
  function train (line 42) | def train():
  function test (line 111) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-coin/edgebank.py
  function test (line 34) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 79) | def get_args():

FILE: examples/linkproppred/tgbl-coin/tgn.py
  function train (line 44) | def train():
  function test (line 112) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-comment/dyrep.py
  function train (line 43) | def train():
  function test_one_vs_many (line 102) | def test_one_vs_many(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-comment/edgebank.py
  function test (line 34) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 79) | def get_args():

FILE: examples/linkproppred/tgbl-comment/tgn.py
  function train (line 44) | def train():
  function test (line 112) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-enron/edgebank.py
  function test (line 32) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 77) | def get_args():

FILE: examples/linkproppred/tgbl-flight/dyrep.py
  function train (line 42) | def train():
  function test (line 111) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-flight/edgebank.py
  function test (line 34) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 79) | def get_args():

FILE: examples/linkproppred/tgbl-flight/tgn.py
  function train (line 44) | def train():
  function test (line 112) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-lastfm/edgebank.py
  function test (line 32) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 77) | def get_args():

FILE: examples/linkproppred/tgbl-lastfm/tgn.py
  function train (line 41) | def train():
  function test (line 109) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-review/dyrep.py
  function train (line 42) | def train():
  function test (line 111) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-review/edgebank.py
  function test (line 32) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 77) | def get_args():

FILE: examples/linkproppred/tgbl-review/tgn.py
  function train (line 44) | def train():
  function test (line 112) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-subreddit/edgebank.py
  function test (line 32) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 77) | def get_args():

FILE: examples/linkproppred/tgbl-subreddit/tgn.py
  function train (line 42) | def train():
  function test (line 110) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-uci/edgebank.py
  function test (line 32) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 77) | def get_args():

FILE: examples/linkproppred/tgbl-wiki/dyrep.py
  function train (line 42) | def train():
  function test (line 111) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tgbl-wiki/edgebank.py
  function test (line 32) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 77) | def get_args():

FILE: examples/linkproppred/tgbl-wiki/tgn.py
  function train (line 43) | def train():
  function test (line 111) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-forum/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/thgl-forum/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/thgl-forum/sthn.py
  function print_model_info (line 72) | def print_model_info(model):
  function get_args (line 78) | def get_args():
  function load_model (line 122) | def load_model(args):
  function load_graph (line 160) | def load_graph(data):
  function load_all_data (line 211) | def load_all_data(args):
  function test (line 276) | def test(data, test_mask, model, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-forum/tgn.py
  function train (line 42) | def train():
  function test (line 110) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-github/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/thgl-github/recurrencybaseline.py
  function predict (line 35) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 86) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 129) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 165) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 257) | def get_args():

FILE: examples/linkproppred/thgl-github/sthn.py
  function print_model_info (line 72) | def print_model_info(model):
  function get_args (line 78) | def get_args():
  function load_model (line 122) | def load_model(args):
  function load_graph (line 160) | def load_graph(data):
  function load_all_data (line 211) | def load_all_data(args):
  function test (line 276) | def test(data, test_mask, model, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-github/tgn.py
  function train (line 46) | def train():
  function test (line 114) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-myket/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/thgl-myket/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/thgl-myket/sthn.py
  function print_model_info (line 72) | def print_model_info(model):
  function get_args (line 78) | def get_args():
  function load_model (line 122) | def load_model(args):
  function load_graph (line 160) | def load_graph(data):
  function load_all_data (line 211) | def load_all_data(args):
  function test (line 276) | def test(data, test_mask, model, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-myket/tgn.py
  function train (line 42) | def train():
  function test (line 110) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-software/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/thgl-software/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/thgl-software/sthn.py
  function print_model_info (line 72) | def print_model_info(model):
  function get_args (line 78) | def get_args():
  function load_model (line 122) | def load_model(args):
  function load_graph (line 160) | def load_graph(data):
  function load_all_data (line 211) | def load_all_data(args):
  function test (line 276) | def test(data, test_mask, model, neg_sampler, split_mode):

FILE: examples/linkproppred/thgl-software/tgn.py
  function train (line 42) | def train():
  function test (line 110) | def test(loader, neg_sampler, split_mode):

FILE: examples/linkproppred/tkgl-icews/cen.py
  function test (line 28) | def test(model, history_len, history_list, test_list, num_rels, num_node...
  function run_experiment (line 105) | def run_experiment(args, trainvalidtest_id=0, n_hidden=None, n_layers=No...

FILE: examples/linkproppred/tkgl-icews/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/tkgl-icews/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/tkgl-icews/regcn.py
  function test (line 29) | def test(model, history_list, test_list, num_rels, num_nodes, use_cuda, ...
  function run_experiment (line 106) | def run_experiment(args, n_hidden=None, n_layers=None, dropout=None, n_b...

FILE: examples/linkproppred/tkgl-icews/timetraveler.py
  class QuadruplesDataset (line 35) | class QuadruplesDataset(Dataset):
    method __init__ (line 38) | def __init__(self, examples):
    method __len__ (line 46) | def __len__(self):
    method __getitem__ (line 49) | def __getitem__(self, item):
  function set_logger (line 56) | def set_logger(save_path):
  function preprocess_data (line 76) | def preprocess_data(args, config, timestamps, save_path, all_quads):
  function log_metrics (line 97) | def log_metrics(mode, step, metrics):
  function main (line 102) | def main(args):

FILE: examples/linkproppred/tkgl-icews/tlogic.py
  function learn_rules (line 28) | def learn_rules(i, num_relations):
  function apply_rules (line 75) | def apply_rules(i, num_queries, rules_dict, neg_sampler, data, window, l...
  function get_args (line 230) | def get_args():

FILE: examples/linkproppred/tkgl-polecat/cen.py
  function test (line 28) | def test(model, history_len, history_list, test_list, num_rels, num_node...
  function run_experiment (line 105) | def run_experiment(args, trainvalidtest_id=0, n_hidden=None, n_layers=No...

FILE: examples/linkproppred/tkgl-polecat/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/tkgl-polecat/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/tkgl-polecat/regcn.py
  function test (line 29) | def test(model, history_list, test_list, num_rels, num_nodes, use_cuda, ...
  function run_experiment (line 106) | def run_experiment(args, n_hidden=None, n_layers=None, dropout=None, n_b...

FILE: examples/linkproppred/tkgl-polecat/timetraveler.py
  class QuadruplesDataset (line 35) | class QuadruplesDataset(Dataset):
    method __init__ (line 38) | def __init__(self, examples):
    method __len__ (line 46) | def __len__(self):
    method __getitem__ (line 49) | def __getitem__(self, item):
  function set_logger (line 56) | def set_logger(save_path):
  function preprocess_data (line 76) | def preprocess_data(args, config, timestamps, save_path, all_quads):
  function log_metrics (line 97) | def log_metrics(mode, step, metrics):
  function main (line 102) | def main(args):

FILE: examples/linkproppred/tkgl-polecat/tlogic.py
  function learn_rules (line 28) | def learn_rules(i, num_relations):
  function apply_rules (line 75) | def apply_rules(i, num_queries, rules_dict, neg_sampler, data, window, l...
  function get_args (line 230) | def get_args():

FILE: examples/linkproppred/tkgl-smallpedia/cen.py
  function test (line 28) | def test(model, history_len, history_list, test_list, num_rels, num_node...
  function run_experiment (line 105) | def run_experiment(args, trainvalidtest_id=0, n_hidden=None, n_layers=No...

FILE: examples/linkproppred/tkgl-smallpedia/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/tkgl-smallpedia/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/tkgl-smallpedia/regcn.py
  function test (line 29) | def test(model, history_list, test_list, num_rels, num_nodes, use_cuda, ...
  function run_experiment (line 106) | def run_experiment(args, n_hidden=None, n_layers=None, dropout=None, n_b...

FILE: examples/linkproppred/tkgl-smallpedia/timetraveler.py
  class QuadruplesDataset (line 35) | class QuadruplesDataset(Dataset):
    method __init__ (line 38) | def __init__(self, examples):
    method __len__ (line 46) | def __len__(self):
    method __getitem__ (line 49) | def __getitem__(self, item):
  function set_logger (line 56) | def set_logger(save_path):
  function preprocess_data (line 76) | def preprocess_data(args, config, timestamps, save_path, all_quads):
  function log_metrics (line 97) | def log_metrics(mode, step, metrics):
  function main (line 102) | def main(args):

FILE: examples/linkproppred/tkgl-smallpedia/tlogic.py
  function learn_rules (line 28) | def learn_rules(i, num_relations):
  function apply_rules (line 75) | def apply_rules(i, num_queries, rules_dict, neg_sampler, data, window, l...
  function get_args (line 230) | def get_args():

FILE: examples/linkproppred/tkgl-wikidata/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/tkgl-wikidata/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/tkgl-wikidata/regcn.py
  function test (line 29) | def test(model, history_list, test_list, num_rels, num_nodes, use_cuda, ...
  function run_experiment (line 106) | def run_experiment(args, n_hidden=None, n_layers=None, dropout=None, n_b...

FILE: examples/linkproppred/tkgl-wikidata/tlogic.py
  function learn_rules (line 28) | def learn_rules(i, num_relations):
  function apply_rules (line 75) | def apply_rules(i, num_queries, rules_dict, neg_sampler, data, window, l...
  function get_args (line 230) | def get_args():

FILE: examples/linkproppred/tkgl-yago/cen.py
  function test (line 28) | def test(model, history_len, history_list, test_list, num_rels, num_node...
  function run_experiment (line 105) | def run_experiment(args, trainvalidtest_id=0, n_hidden=None, n_layers=No...

FILE: examples/linkproppred/tkgl-yago/edgebank.py
  function test (line 36) | def test(data, test_mask, neg_sampler, split_mode):
  function get_args (line 86) | def get_args():

FILE: examples/linkproppred/tkgl-yago/recurrencybaseline.py
  function predict (line 34) | def predict(num_processes,  data_c_rel, all_data_c_rel, alpha, lmbda_psi,
  function test (line 85) | def test(best_config, all_relations,test_data_prel, all_data_prel, neg_s...
  function read_dict_compute_mrr (line 128) | def read_dict_compute_mrr(split_mode='test'):
  function train (line 164) | def train(params_dict, rels,val_data_prel, trainval_data_prel, neg_sampl...
  function get_args (line 256) | def get_args():

FILE: examples/linkproppred/tkgl-yago/regcn.py
  function test (line 29) | def test(model, history_list, test_list, num_rels, num_nodes, use_cuda, ...
  function run_experiment (line 106) | def run_experiment(args, n_hidden=None, n_layers=None, dropout=None, n_b...

FILE: examples/linkproppred/tkgl-yago/timetraveler.py
  class QuadruplesDataset (line 35) | class QuadruplesDataset(Dataset):
    method __init__ (line 38) | def __init__(self, examples):
    method __len__ (line 46) | def __len__(self):
    method __getitem__ (line 49) | def __getitem__(self, item):
  function set_logger (line 56) | def set_logger(save_path):
  function preprocess_data (line 76) | def preprocess_data(args, config, timestamps, save_path, all_quads):
  function log_metrics (line 97) | def log_metrics(mode, step, metrics):
  function main (line 102) | def main(args):

FILE: examples/linkproppred/tkgl-yago/tlogic.py
  function learn_rules (line 28) | def learn_rules(i, num_relations):
  function apply_rules (line 75) | def apply_rules(i, num_queries, rules_dict, neg_sampler, data, window, l...
  function get_args (line 230) | def get_args():

FILE: examples/nodeproppred/tgbn-genre/dyrep.py
  function process_edges (line 29) | def process_edges(src, dst, t, msg):
  function train (line 39) | def train():
  function test (line 139) | def test(loader):

FILE: examples/nodeproppred/tgbn-genre/moving_average.py
  function test_n_upate (line 41) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-genre/persistant_forecast.py
  function test_n_upate (line 52) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-genre/tgn.py
  function plot_curve (line 100) | def plot_curve(scores, out_name):
  function process_edges (line 107) | def process_edges(src, dst, t, msg):
  function train (line 114) | def train():
  function test (line 215) | def test(loader):

FILE: examples/nodeproppred/tgbn-reddit/dyrep.py
  function process_edges (line 31) | def process_edges(src, dst, t, msg):
  function train (line 41) | def train():
  function test (line 141) | def test(loader):

FILE: examples/nodeproppred/tgbn-reddit/moving_average.py
  function test_n_upate (line 41) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-reddit/persistant_forecast.py
  function test_n_upate (line 52) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-reddit/tgn.py
  function plot_curve (line 101) | def plot_curve(scores, out_name):
  function process_edges (line 108) | def process_edges(src, dst, t, msg):
  function train (line 115) | def train():
  function test (line 215) | def test(loader):

FILE: examples/nodeproppred/tgbn-token/dyrep.py
  function process_edges (line 31) | def process_edges(src, dst, t, msg):
  function train (line 41) | def train():
  function test (line 141) | def test(loader):

FILE: examples/nodeproppred/tgbn-token/moving_average.py
  function test_n_upate (line 42) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-token/persistant_forecast.py
  function test_n_upate (line 52) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-token/tgn.py
  function plot_curve (line 102) | def plot_curve(scores, out_name):
  function process_edges (line 109) | def process_edges(src, dst, t, msg):
  function train (line 116) | def train():
  function test (line 215) | def test(loader):

FILE: examples/nodeproppred/tgbn-trade/count_new_nodes.py
  function count_nodes (line 23) | def count_nodes(data, test_mask, nodebank):
  function get_args (line 71) | def get_args():

FILE: examples/nodeproppred/tgbn-trade/dyrep.py
  function process_edges (line 29) | def process_edges(src, dst, t, msg):
  function train (line 39) | def train():
  function test (line 139) | def test(loader):

FILE: examples/nodeproppred/tgbn-trade/moving_average.py
  function test_n_upate (line 43) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-trade/persistant_forecast.py
  function test_n_upate (line 42) | def test_n_upate(loader):

FILE: examples/nodeproppred/tgbn-trade/tgn.py
  function plot_curve (line 100) | def plot_curve(scores, out_name):
  function process_edges (line 107) | def process_edges(src, dst, t, msg):
  function train (line 114) | def train():
  function test (line 215) | def test(loader):

FILE: modules/decoder.py
  class LinkPredictor (line 12) | class LinkPredictor(torch.nn.Module):
    method __init__ (line 18) | def __init__(self, in_channels):
    method forward (line 24) | def forward(self, z_src, z_dst):
  class NodePredictor (line 30) | class NodePredictor(torch.nn.Module):
    method __init__ (line 31) | def __init__(self, in_dim, out_dim):
    method forward (line 36) | def forward(self, node_embed):
  class ConvTransE (line 45) | class ConvTransE(torch.nn.Module):
    method __init__ (line 49) | def __init__(self, num_entities, embedding_dim, input_dropout=0, hidde...
    method forward (line 75) | def forward(self, embedding, emb_rel, triplets, partial_embeding=None,...
    method forward_inner (line 95) | def forward_inner(self, embedding, emb_rel, triplets, idx=0, partial_e...

FILE: modules/early_stopping.py
  class EarlyStopMonitor (line 10) | class EarlyStopMonitor(object):
    method __init__ (line 12) | def __init__(self, save_model_dir: str, save_model_id: str,
    method get_best_model_path (line 38) | def get_best_model_path(self):
    method step_check (line 44) | def step_check(self, curr_metric: float, models_dict: dict):
    method save_checkpoint (line 68) | def save_checkpoint(self, models_dict: dict):
    method load_checkpoint (line 80) | def load_checkpoint(self, models_dict: dict):

FILE: modules/edgebank_predictor.py
  class EdgeBankPredictor (line 13) | class EdgeBankPredictor(object):
    method __init__ (line 14) | def __init__(
    method update_memory (line 52) | def update_memory(self,
    method start_time (line 72) | def start_time(self) -> int:
    method end_time (line 83) | def end_time(self) -> int:
    method _update_unlimited_memory (line 93) | def _update_unlimited_memory(self,
    method _update_time_window_memory (line 106) | def _update_time_window_memory(self,
    method predict_link (line 135) | def predict_link(self,

FILE: modules/emb_module.py
  class GraphAttentionEmbedding (line 11) | class GraphAttentionEmbedding(torch.nn.Module):
    method __init__ (line 17) | def __init__(self, in_channels, out_channels, msg_dim, time_enc):
    method forward (line 25) | def forward(self, x, last_update, edge_index, t, msg):
  class TimeEmbedding (line 32) | class TimeEmbedding(torch.nn.Module):
    method __init__ (line 33) | def __init__(self, in_channels, out_channels):
    method forward (line 48) | def forward(self, x, last_update, t):

FILE: modules/heuristics.py
  class PersistantForecaster (line 4) | class PersistantForecaster:
    method __init__ (line 5) | def __init__(self, num_class):
    method update_dict (line 9) | def update_dict(self, node_id, label):
    method query_dict (line 12) | def query_dict(self, node_id):
  class MovingAverage (line 25) | class MovingAverage:
    method __init__ (line 26) | def __init__(self, num_class, window=7):
    method update_dict (line 31) | def update_dict(self, node_id, label):
    method query_dict (line 38) | def query_dict(self, node_id):

FILE: modules/memory_module.py
  class TGNMemory (line 25) | class TGNMemory(torch.nn.Module):
    method __init__ (line 49) | def __init__(
    method device (line 91) | def device(self) -> torch.device:
    method reset_parameters (line 94) | def reset_parameters(self):
    method reset_state (line 106) | def reset_state(self):
    method detach (line 112) | def detach(self):
    method forward (line 116) | def forward(self, n_id: Tensor) -> Tuple[Tensor, Tensor]:
    method update_state (line 126) | def update_state(self, src: Tensor, dst: Tensor, t: Tensor, raw_msg: T...
    method _reset_message_store (line 140) | def _reset_message_store(self):
    method _update_memory (line 147) | def _update_memory(self, n_id: Tensor):
    method _get_updated_memory (line 152) | def _get_updated_memory(self, n_id: Tensor) -> Tuple[Tensor, Tensor]:
    method _update_msg_store (line 180) | def _update_msg_store(
    method _compute_msg (line 193) | def _compute_msg(
    method train (line 209) | def train(self, mode: bool = True):
  class DyRepMemory (line 218) | class DyRepMemory(torch.nn.Module):
    method __init__ (line 242) | def __init__(self, num_nodes: int, raw_msg_dim: int, memory_dim: int,
    method device (line 281) | def device(self) -> torch.device:
    method reset_parameters (line 284) | def reset_parameters(self):
    method reset_state (line 296) | def reset_state(self):
    method detach (line 302) | def detach(self):
    method forward (line 306) | def forward(self, n_id: Tensor) -> Tuple[Tensor, Tensor]:
    method update_state (line 316) | def update_state(self, src: Tensor, dst: Tensor, t: Tensor, raw_msg: T...
    method _reset_message_store (line 331) | def _reset_message_store(self):
    method _update_memory (line 338) | def _update_memory(self, n_id: Tensor, embeddings: Tensor = None, asso...
    method _get_updated_memory (line 343) | def _get_updated_memory(self, n_id: Tensor, embeddings: Tensor = None,...
    method _update_msg_store (line 369) | def _update_msg_store(self, src: Tensor, dst: Tensor, t: Tensor,
    method _compute_msg (line 376) | def _compute_msg(self, n_id: Tensor, msg_store: TGNMessageStoreType, m...
    method train (line 414) | def train(self, mode: bool = True):

FILE: modules/msg_agg.py
  class LastAggregator (line 15) | class LastAggregator(torch.nn.Module):
    method forward (line 16) | def forward(self, msg: Tensor, index: Tensor, t: Tensor, dim_size: int):
  class MeanAggregator (line 24) | class MeanAggregator(torch.nn.Module):
    method forward (line 25) | def forward(self, msg: Tensor, index: Tensor, t: Tensor, dim_size: int):

FILE: modules/msg_func.py
  class IdentityMessage (line 12) | class IdentityMessage(torch.nn.Module):
    method __init__ (line 13) | def __init__(self, raw_msg_dim: int, memory_dim: int, time_dim: int):
    method forward (line 17) | def forward(self, z_src: Tensor, z_dst: Tensor, raw_msg: Tensor, t_enc...

FILE: modules/neighbor_loader.py
  class LastNeighborLoader (line 15) | class LastNeighborLoader:
    method __init__ (line 16) | def __init__(self, num_nodes: int, size: int, device=None):
    method __call__ (line 25) | def __call__(self, n_id: Tensor) -> Tuple[Tensor, Tensor, Tensor]:
    method insert (line 41) | def insert(self, src: Tensor, dst: Tensor):
    method reset_state (line 83) | def reset_state(self):

FILE: modules/nodebank.py
  class NodeBank (line 4) | class NodeBank(object):
    method __init__ (line 5) | def __init__(
    method update_memory (line 21) | def update_memory(self,
    method query_node (line 37) | def query_node(self, node: int) -> bool:

FILE: modules/recurrencybaseline_predictor.py
  function baseline_predict_remote (line 20) | def baseline_predict_remote(num_queries, test_data, all_data, window, ba...
  function baseline_predict (line 29) | def baseline_predict(num_queries, test_data, all_data, window, basis_dic...
  function match_body_relations (line 134) | def match_body_relations(rule, edges, test_query_sub):
  function score_delta (line 165) | def score_delta(cands_ts, test_query_ts, lmbda):
  function get_window_edges (line 177) | def get_window_edges(all_data, test_query_ts, window=-2, first_test_quer...
  function quads_per_rel (line 219) | def quads_per_rel(quads):
  function get_candidates_psi (line 237) | def get_candidates_psi(rule_walks, test_query_ts, cands_dict,lmbda, sum_...
  function update_delta_t (line 262) | def update_delta_t(min_ts, max_ts, cur_ts, lmbda):
  function score_psi (line 278) | def score_psi(cands_walks, test_query_ts, lmbda, sum_delta_t):
  function update_distributions (line 304) | def update_distributions(ts_edges,num_rels):
  function calculate_obj_distribution (line 311) | def calculate_obj_distribution(edges, num_rels):
  function update_delta_t (line 334) | def update_delta_t(min_ts, max_ts, cur_ts, lmbda):

FILE: modules/rgcn_layers.py
  class RGCNLayer (line 11) | class RGCNLayer(nn.Module):
    method __init__ (line 12) | def __init__(self, in_feat, out_feat, bias=None, activation=None,
    method propagate (line 52) | def propagate(self, g):
    method forward (line 55) | def forward(self, g, prev_h=[]):
  class RGCNBasisLayer (line 101) | class RGCNBasisLayer(RGCNLayer):
    method __init__ (line 102) | def __init__(self, in_feat, out_feat, num_rels, num_bases=-1, bias=None,
    method propagate (line 125) | def propagate(self, g):
  class RGCNBlockLayer (line 154) | class RGCNBlockLayer(RGCNLayer):
    method __init__ (line 155) | def __init__(self, in_feat, out_feat, num_rels, num_bases, bias=None,
    method msg_func (line 174) | def msg_func(self, edges):
    method propagate (line 181) | def propagate(self, g):
    method apply_func (line 185) | def apply_func(self, nodes):
  class UnionRGCNLayer (line 189) | class UnionRGCNLayer(nn.Module):
    method __init__ (line 190) | def __init__(self, in_feat, out_feat, num_rels, num_bases=-1,  bias=None,
    method propagate (line 226) | def propagate(self, g):
    method forward (line 229) | def forward(self, g, prev_h):
    method msg_func (line 263) | def msg_func(self, edges):
    method apply_func (line 284) | def apply_func(self, nodes):

FILE: modules/rgcn_model.py
  class BaseRGCN (line 8) | class BaseRGCN(nn.Module):
    method __init__ (line 9) | def __init__(self, num_nodes, h_dim, out_dim, num_rels, num_bases=-1, ...
    method build_model (line 35) | def build_model(self):
    method create_features (line 52) | def create_features(self):
    method build_input_layer (line 55) | def build_input_layer(self):
    method build_hidden_layer (line 58) | def build_hidden_layer(self, idx):
    method build_output_layer (line 61) | def build_output_layer(self):
    method forward (line 64) | def forward(self, g):

FILE: modules/rrgcn.py
  class RGCNCell (line 13) | class RGCNCell(BaseRGCN):
    method build_hidden_layer (line 14) | def build_hidden_layer(self, idx):
    method forward (line 30) | def forward(self, g, init_ent_emb):
  class RecurrentRGCNCEN (line 52) | class RecurrentRGCNCEN(nn.Module):
    method __init__ (line 53) | def __init__(self, decoder_name, encoder_name, num_ents, num_rels, h_d...
    method forward (line 109) | def forward(self, g_list, use_cuda):
    method predict (line 122) | def predict(self, test_graph, test_triplets, use_cuda, neg_samples_bat...
    method get_ft_loss (line 165) | def get_ft_loss(self, glist, triple_list,  use_cuda):
    method get_loss (line 186) | def get_loss(self, glist, triples, prev_model, use_cuda):
  class RecurrentRGCNREGCN (line 208) | class RecurrentRGCNREGCN(nn.Module):
    method __init__ (line 209) | def __init__(self, decoder_name, encoder_name, num_ents, num_rels, num...
    method forward (line 293) | def forward(self, g_list, static_graph, use_cuda):
    method predict (line 336) | def predict(self, test_graph, num_rels, static_graph, test_triplets, u...
    method get_mask_nonzero (line 370) | def get_mask_nonzero(self, static_embedding):
    method get_loss (line 377) | def get_loss(self, glist, triples, static_graph, use_cuda):

FILE: modules/sampler_core.cpp
  class TemporalGraphBlock (line 17) | class TemporalGraphBlock
    method TemporalGraphBlock (line 33) | TemporalGraphBlock() {}
    method TemporalGraphBlock (line 34) | TemporalGraphBlock(std::vector<NodeIDType> &_row, std::vector<NodeIDTy...
  class ParallelSampler (line 41) | class ParallelSampler
    method ParallelSampler (line 63) | ParallelSampler(std::vector<EdgeIDType> &_indptr, std::vector<EdgeIDTy...
    method reset (line 89) | void reset()
    method update_ts_ptr (line 100) | void update_ts_ptr(int slc, std::vector<NodeIDType> &root_nodes,
    method add_neighbor (line 126) | inline void add_neighbor(std::vector<NodeIDType> *_row, std::vector<No...
    method combine_coo (line 150) | inline void combine_coo(TemporalGraphBlock &_ret, std::vector<NodeIDTy...
    method sample_layer (line 200) | void sample_layer(std::vector<NodeIDType> &_root_nodes, std::vector<Ti...
    method sample (line 336) | void sample(std::vector<NodeIDType> &root_nodes, std::vector<TimeStamp...
  function vec2npy (line 362) | inline py::array vec2npy(const std::vector<T> &vec)
  function PYBIND11_MODULE (line 373) | PYBIND11_MODULE(sampler_core, m)

FILE: modules/sthn.py
  function set_seed (line 31) | def set_seed(seed):
  function row_norm (line 37) | def row_norm(adj_t):
  class NegLinkSampler (line 61) | class NegLinkSampler:
    method __init__ (line 65) | def __init__(self, num_nodes):
    method sample (line 68) | def sample(self, n):
  function get_parallel_sampler (line 71) | def get_parallel_sampler(g, num_neighbors=10):
  function get_mini_batch (line 101) | def get_mini_batch(sampler, root_nodes, ts, num_hops): # neg_samples is ...
  function fetch_subgraph (line 113) | def fetch_subgraph(sampler, root_node, root_time, num_hops):
  function construct_mini_batch_giant_graph (line 181) | def construct_mini_batch_giant_graph(all_graphs, max_num_edges):
  function print_subgraph_data (line 255) | def print_subgraph_data(subgraph_data):
  class SubgraphSampler (line 278) | class SubgraphSampler:
    method __init__ (line 279) | def __init__(self, all_root_nodes, all_ts, sampler, args):
    method mini_batch (line 285) | def mini_batch(self, ind, mini_batch_inds):
  function get_subgraph_sampler (line 290) | def get_subgraph_sampler(args, g, df, mode):
  function pre_compute_subgraphs (line 342) | def pre_compute_subgraphs(args, g, df, mode, negative_sampler=None, spli...
  function get_random_inds (line 429) | def get_random_inds(num_subgraph, cached_neg_samples, neg_samples):
  function get_all_inds (line 442) | def get_all_inds(num_subgraph, neg_samples):
  function check_data_leakage (line 454) | def check_data_leakage(args, g, df):
  function get_inputs_for_ind (line 492) | def get_inputs_for_ind(subgraphs, mode, cached_neg_samples, neg_samples,...
  function run (line 555) | def run(model, optimizer, args, subgraphs, df, node_feats, edge_feats, M...
  function link_pred_train (line 616) | def link_pred_train(model, args, g, df, node_feats, edge_feats):
  function compute_sign_feats (line 696) | def compute_sign_feats(node_feats, df, start_i, num_links, root_nodes, a...
  function get_emb (line 750) | def get_emb(sin_inp):
  class PositionalEncoding1D (line 758) | class PositionalEncoding1D(nn.Module):
    method __init__ (line 759) | def __init__(self, channels):
    method forward (line 771) | def forward(self, tensor):
  class PositionalEncodingPermute1D (line 794) | class PositionalEncodingPermute1D(nn.Module):
    method __init__ (line 795) | def __init__(self, channels):
    method forward (line 802) | def forward(self, tensor):
    method org_channels (line 808) | def org_channels(self):
  class PositionalEncoding2D (line 812) | class PositionalEncoding2D(nn.Module):
    method __init__ (line 813) | def __init__(self, channels):
    method forward (line 825) | def forward(self, tensor):
  class PositionalEncodingPermute2D (line 854) | class PositionalEncodingPermute2D(nn.Module):
    method __init__ (line 855) | def __init__(self, channels):
    method forward (line 862) | def forward(self, tensor):
    method org_channels (line 868) | def org_channels(self):
  class PositionalEncoding3D (line 872) | class PositionalEncoding3D(nn.Module):
    method __init__ (line 873) | def __init__(self, channels):
    method forward (line 887) | def forward(self, tensor):
  class PositionalEncodingPermute3D (line 920) | class PositionalEncodingPermute3D(nn.Module):
    method __init__ (line 921) | def __init__(self, channels):
    method forward (line 928) | def forward(self, tensor):
    method org_channels (line 934) | def org_channels(self):
  class Summer (line 938) | class Summer(nn.Module):
    method __init__ (line 939) | def __init__(self, penc):
    method forward (line 946) | def forward(self, tensor):
  class TimeEncode (line 970) | class TimeEncode(nn.Module):
    method __init__ (line 975) | def __init__(self, dim):
    method reset_parameters (line 981) | def reset_parameters(self, ):
    method forward (line 989) | def forward(self, t):
  class FeedForward (line 1002) | class FeedForward(nn.Module):
    method __init__ (line 1006) | def __init__(self, dims, expansion_factor, dropout=0, use_single_layer...
    method reset_parameters (line 1023) | def reset_parameters(self):
    method forward (line 1028) | def forward(self, x):
  class TransformerBlock (line 1039) | class TransformerBlock(nn.Module):
    method __init__ (line 1044) | def __init__(self, dims,
    method reset_parameters (line 1066) | def reset_parameters(self):
    method token_mixer (line 1073) | def token_mixer(self, x):
    method channel_mixer (line 1077) | def channel_mixer(self, x):
    method forward (line 1082) | def forward(self, x):
  class _MultiheadAttention (line 1090) | class _MultiheadAttention(nn.Module):
    method __init__ (line 1091) | def __init__(self, d_model, n_heads, d_k=None, d_v=None, res_attention...
    method reset_parameters (line 1115) | def reset_parameters(self):
    method forward (line 1121) | def forward(self, Q:Tensor, K:Optional[Tensor]=None, V:Optional[Tensor...
  class _ScaledDotProductAttention (line 1143) | class _ScaledDotProductAttention(nn.Module):
    method __init__ (line 1148) | def __init__(self, d_model, n_heads, attn_dropout=0., res_attention=Fa...
    method forward (line 1156) | def forward(self, q:Tensor, k:Tensor, v:Tensor, prev:Optional[Tensor]=...
  class FeatEncode (line 1200) | class FeatEncode(nn.Module):
    method __init__ (line 1204) | def __init__(self, time_dims, feat_dims, out_dims):
    method reset_parameters (line 1211) | def reset_parameters(self):
    method forward (line 1215) | def forward(self, edge_feats, edge_ts):
  class Patch_Encoding (line 1220) | class Patch_Encoding(nn.Module):
    method __init__ (line 1225) | def __init__(self, per_graph_size, time_channels,
    method reset_parameters (line 1261) | def reset_parameters(self):
    method forward (line 1268) | def forward(self, edge_feats, edge_ts, batch_size, inds):
  class EdgePredictor_per_node (line 1293) | class EdgePredictor_per_node(torch.nn.Module):
    method __init__ (line 1298) | def __init__(self, dim_in_time, dim_in_node, predict_class):
    method reset_parameters (line 1311) | def reset_parameters(self, ):
    method forward (line 1316) | def forward(self, h, neg_samples=1):
  class STHN_Interface (line 1328) | class STHN_Interface(nn.Module):
    method __init__ (line 1329) | def __init__(self, mlp_mixer_configs, edge_predictor_configs):
    method reset_parameters (line 1342) | def reset_parameters(self):
    method forward (line 1347) | def forward(self, model_inputs, neg_samples, node_feats):
    method predict (line 1355) | def predict(self, model_inputs, neg_samples, node_feats):
  class Multiclass_Interface (line 1369) | class Multiclass_Interface(nn.Module):
    method __init__ (line 1370) | def __init__(self, mlp_mixer_configs, edge_predictor_configs):
    method reset_parameters (line 1383) | def reset_parameters(self):
    method forward (line 1388) | def forward(self, model_inputs, neg_samples, node_feats):
    method predict (line 1399) | def predict(self, model_inputs, neg_samples, node_feats):

FILE: modules/time_enc.py
  class TimeEncoder (line 14) | class TimeEncoder(torch.nn.Module):
    method __init__ (line 15) | def __init__(self, out_channels: int):
    method reset_parameters (line 20) | def reset_parameters(self):
    method forward (line 23) | def forward(self, t: Tensor) -> Tensor:

FILE: modules/timetraveler_agent.py
  class HistoryEncoder (line 14) | class HistoryEncoder(nn.Module):
    method __init__ (line 15) | def __init__(self, config):
    method set_hiddenx (line 21) | def set_hiddenx(self, batch_size):
    method forward (line 30) | def forward(self, prev_action, mask):
  class PolicyMLP (line 37) | class PolicyMLP(nn.Module):
    method __init__ (line 38) | def __init__(self, config):
    method forward (line 43) | def forward(self, state_query):
  class DynamicEmbedding (line 48) | class DynamicEmbedding(nn.Module):
    method __init__ (line 49) | def __init__(self, n_ent, dim_ent, dim_t):
    method forward (line 55) | def forward(self, entities, dt):
  class StaticEmbedding (line 67) | class StaticEmbedding(nn.Module):
    method __init__ (line 68) | def __init__(self, n_ent, dim_ent):
    method forward (line 72) | def forward(self, entities, timestamps=None):
  class Agent (line 75) | class Agent(nn.Module):
    method __init__ (line 76) | def __init__(self, config):
    method forward (line 101) | def forward(self, prev_relation, current_entities, current_timestamps,
    method get_im_embedding (line 164) | def get_im_embedding(self, cooccurrence_entities):
    method update_entity_embedding (line 173) | def update_entity_embedding(self, entity, ims, mu):
    method entities_embedding_shift (line 182) | def entities_embedding_shift(self, entity, im, mu):
    method back_entities_embedding (line 187) | def back_entities_embedding(self, entity):

FILE: modules/timetraveler_dirichlet.py
  class NotConvergingError (line 54) | class NotConvergingError(Exception):
  function test (line 60) | def test(D1, D2, method="meanprecision", maxiter=None):
  function pdf (line 104) | def pdf(alphas):
  function meanprecision (line 136) | def meanprecision(a):
  function loglikelihood (line 154) | def loglikelihood(D, a):
  function mle (line 172) | def mle(D, tol=1e-7, method="meanprecision", maxiter=None):
  function _fixedpoint (line 201) | def _fixedpoint(D, tol=1e-7, maxiter=None):
  function _meanprecision (line 236) | def _meanprecision(D, tol=1e-7, maxiter=None):
  function _fit_s (line 285) | def _fit_s(D, a0, logp, tol=1e-7, maxiter=1000):
  function _fit_m (line 332) | def _fit_m(D, a0, logp, tol=1e-7, maxiter=1000):
  function _init_a (line 365) | def _init_a(D):
  function _ipsi (line 381) | def _ipsi(y, tol=1.48e-9, maxiter=10):
  function _trigamma (line 414) | def _trigamma(x):
  class MLE_Dirchlet (line 418) | class MLE_Dirchlet(object):
    method __init__ (line 419) | def __init__(self, trainQuads, num_r, k, timespan,
    method get_entity_occ_times (line 441) | def get_entity_occ_times(self, trainQuads):
    method get_relations_observed_data (line 454) | def get_relations_observed_data(self, trainQuads):
    method mle_dirchlet (line 479) | def mle_dirchlet(self):
  class Dirichlet (line 488) | class Dirichlet(object):
    method __init__ (line 489) | def __init__(self, alphas, k):
    method __call__ (line 498) | def __call__(self, rel, dt):

FILE: modules/timetraveler_environment.py
  class Env (line 14) | class Env(object):
    method __init__ (line 15) | def __init__(self, examples, config, state_action_space=None):
    method build_graph (line 33) | def build_graph(self, examples):
    method get_state_actions_space_complete (line 63) | def get_state_actions_space_complete(self, entity, time, current_=Fals...
    method next_actions (line 96) | def next_actions(self, entites, times, query_times, max_action_num=200...
    method get_padd_actions (line 123) | def get_padd_actions(self, entites, times, query_times, max_action_num...

FILE: modules/timetraveler_episode.py
  class Episode (line 12) | class Episode(nn.Module):
    method __init__ (line 13) | def __init__(self, env, agent, config):
    method forward (line 22) | def forward(self, query_entities, query_timestamps, query_relations):
    method beam_search (line 85) | def beam_search(self, query_entities, query_timestamps, query_relations):

FILE: modules/timetraveler_policygradient.py
  class ReactiveBaseline (line 14) | class ReactiveBaseline(object):
    method __init__ (line 15) | def __init__(self, config, update_rate):
    method get_baseline_value (line 21) | def get_baseline_value(self):
    method update (line 24) | def update(self, target):
  class PG (line 27) | class PG(object):
    method __init__ (line 28) | def __init__(self, config):
    method get_reward (line 35) | def get_reward(self, current_entites, answers):
    method calc_cum_discounted_reward (line 41) | def calc_cum_discounted_reward(self, rewards):
    method entropy_reg_loss (line 54) | def entropy_reg_loss(self, all_logits):
    method calc_reinforce_loss (line 59) | def calc_reinforce_loss(self, all_loss, all_logits, cum_discounted_rew...

FILE: modules/timetraveler_trainertester.py
  class Trainer (line 7) | class Trainer(object):
    method __init__ (line 8) | def __init__(self, model, pg, optimizer, args, distribution=None):
    method train_epoch (line 15) | def train_epoch(self, dataloader, ntriple):
    method save_model (line 66) | def save_model(self, save_path, checkpoint_path='checkpoint.pth'):
  class Tester (line 79) | class Tester(object):
    method __init__ (line 80) | def __init__(self, model, args, train_entities, RelEntCooccurrence, me...
    method get_rank (line 88) | def get_rank(self, score, answer, entities_space, num_ent):
    method test (line 106) | def test(self, dataloader, ntriple, num_nodes, neg_sampler, evaluator,...
  function getRelEntCooccurrence (line 247) | def getRelEntCooccurrence(quadruples, num_rels):

FILE: modules/tkg_utils.py
  function get_args_timetraveler (line 9) | def get_args_timetraveler(args=None):
  function get_model_config_timetraveler (line 78) | def get_model_config_timetraveler(args, num_ent, num_rel):
  function get_args_cen (line 103) | def get_args_cen():
  function get_args_regcn (line 222) | def get_args_regcn():
  function compute_min_distance (line 339) | def compute_min_distance(unique_sorted_timestamps):
  function compute_maxminmean_distances (line 347) | def compute_maxminmean_distances(unique_sorted_timestamps):
  function group_by (line 361) | def group_by(data: np.array, key_idx: int) -> dict:
  function tkg_granularity_lookup (line 374) | def tkg_granularity_lookup(dataset_name, ts_distmean):
  function reformat_ts (line 386) | def reformat_ts(timestamps, dataset_name='tkgl'):
  function get_original_ts (line 412) | def get_original_ts(reformatted_ts, ts_dist, min_ts):
  function create_basis_dict (line 425) | def create_basis_dict(data):
  function get_inv_relation_id (line 444) | def get_inv_relation_id(num_rels):
  function create_scores_array (line 460) | def create_scores_array(predictions_dict, num_nodes):

FILE: modules/tkg_utils_dgl.py
  function build_sub_graph (line 7) | def build_sub_graph(num_nodes, num_rels, triples, use_cuda, gpu, mode='d...
  function r2e (line 51) | def r2e(triplets, num_rels):

FILE: modules/tlogic_apply_modules.py
  function filter_rules (line 14) | def filter_rules(rules_dict, min_conf, min_body_supp, rule_lengths):
  function get_window_edges (line 44) | def get_window_edges(all_data, test_query_ts, learn_edges, window=-1, fi...
  function match_body_relations (line 89) | def match_body_relations(rule, edges, test_query_sub):
  function match_body_relations_complete (line 137) | def match_body_relations_complete(rule, edges, test_query_sub):
  function get_walks (line 180) | def get_walks(rule, walk_edges):
  function get_walks_complete (line 231) | def get_walks_complete(rule, walk_edges):
  function check_var_constraints (line 279) | def check_var_constraints(var_constraints, rule_walks):
  function get_candidates (line 301) | def get_candidates(
  function save_candidates (line 338) | def save_candidates(
  function verbalize_walk (line 367) | def verbalize_walk(walk, data):
  function score1 (line 391) | def score1(rule, c=0):
  function score2 (line 408) | def score2(cands_walks, test_query_ts, lmbda):
  function score_12 (line 429) | def score_12(rule, cands_walks, test_query_ts, lmbda, a):

FILE: modules/tlogic_learn_modules.py
  class Temporal_Walk (line 15) | class Temporal_Walk(object):
    method __init__ (line 16) | def __init__(self, learn_data, inv_relation_id, transition_distr):
    method sample_start_edge (line 37) | def sample_start_edge(self, rel_idx):
    method sample_next_edge (line 53) | def sample_next_edge(self, filtered_edges, cur_ts):
    method transition_step (line 80) | def transition_step(self, cur_node, cur_ts, prev_edge, start_node, ste...
    method sample_walk (line 126) | def sample_walk(self, L, rel_idx):
  function store_neighbors (line 167) | def store_neighbors(quads):
  function store_edges (line 186) | def store_edges(quads):
  class Rule_Learner (line 205) | class Rule_Learner(object):
    method __init__ (line 206) | def __init__(self, edges, id2relation, inv_relation_id, output_dir):
    method create_rule (line 230) | def create_rule(self, walk):
    method define_var_constraints (line 267) | def define_var_constraints(self, entities):
    method estimate_confidence (line 286) | def estimate_confidence(self, rule, num_samples=500):
    method sample_body (line 320) | def sample_body(self, body_rels, var_constraints):
    method calculate_rule_support (line 369) | def calculate_rule_support(self, unique_bodies, head_rel):
    method update_rules_dict (line 396) | def update_rules_dict(self, rule):
    method sort_rules_dict (line 412) | def sort_rules_dict(self):
    method save_rules (line 428) | def save_rules(self, dt, rule_lengths, num_walks, transition_distr, se...
    method save_rules_verbalized (line 453) | def save_rules_verbalized(
  function verbalize_rule (line 483) | def verbalize_rule(rule, id2relation):

FILE: tgb/datasets/ICEWS14/ent2word.py
  function load_index (line 10) | def load_index(input_path):

FILE: tgb/datasets/ICEWS14/icews14.py
  function load_index (line 3) | def load_index(input_path):
  function load_tab_list (line 13) | def load_tab_list(input_path):
  function write2csv (line 22) | def write2csv(rows, output_path):
  function main (line 29) | def main():

FILE: tgb/datasets/dataset_scripts/MAG/old/plot_stats.py
  function load_csv (line 5) | def load_csv(fname: str):

FILE: tgb/datasets/dataset_scripts/dgraph.py
  function main (line 39) | def main():

FILE: tgb/datasets/dataset_scripts/process_arxiv.py
  function load_full_json (line 8) | def load_full_json(fname):
  function main (line 20) | def main():

FILE: tgb/datasets/dataset_scripts/process_github.py
  function str_to_timestamp (line 33) | def str_to_timestamp(time_str):
  function parse_issue_comment_events (line 38) | def parse_issue_comment_events(event):
  function parse_issue_event (line 61) | def parse_issue_event(event):
  function parse_pull_request_event (line 90) | def parse_pull_request_event(event):
  function parse_pull_request_review_comment_event (line 119) | def parse_pull_request_review_comment_event(event):
  function parse_fork_event (line 142) | def parse_fork_event(event):
  function parse_member_event (line 156) | def parse_member_event(event):
  function parse_event (line 180) | def parse_event(event):
  function parse_file (line 191) | def parse_file(filename):

FILE: tgb/datasets/dataset_scripts/tgbl-coin.py
  function analyze_csv (line 9) | def analyze_csv(fname):
  function extract_node_dict (line 91) | def extract_node_dict(fname, freq=10):
  function clean_edgelist (line 127) | def clean_edgelist(fname, outname, node_dict):
  function sort_edgelist (line 149) | def sort_edgelist(in_file, outname):
  function main (line 181) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-coin_neg_generator.py
  function main (line 9) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-comment.py
  function find_filenames (line 7) | def find_filenames(path_to_dir):
  function read_edgelist (line 17) | def read_edgelist(fname, outfname, write_header=False):
  function read_nodeattr (line 47) | def read_nodeattr(fname, outfname, write_header=False):
  function combine_edgelist_edgefeat (line 91) | def combine_edgelist_edgefeat(edgefname, featfname, outname):
  function main (line 165) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-comment_neg_generator.py
  function main (line 9) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-flight.py
  function find_csv_filenames (line 9) | def find_csv_filenames(path_to_dir, suffix=".csv"):
  function flight2edgelist (line 20) | def flight2edgelist(
  function load_icao_airports (line 97) | def load_icao_airports(fname="airport_codes.csv"):
  function merge_edgelist (line 118) | def merge_edgelist(input_names: str, in_dir: str, outname: str):
  function clean_node_feat (line 148) | def clean_node_feat(in_file, outname):
  function sort_edgelist (line 188) | def sort_edgelist(in_file, outname):
  function date2ts (line 221) | def date2ts(date_str: str) -> float:
  function main (line 230) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-flight_neg_generator.py
  function main (line 9) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-review.py
  function collect_csv (line 9) | def collect_csv(dir_name="software"):
  function reorder_column (line 15) | def reorder_column(fname: str, outname: str):
  function sort_edgelist (line 36) | def sort_edgelist(fname: str, outname: str):
  function count_degree (line 71) | def count_degree(fname: str):
  function reduce_edgelist (line 97) | def reduce_edgelist(fname: str, outname: str, node10_id: dict):
  function csv_process_review (line 121) | def csv_process_review(
  function main (line 195) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-review_neg_generator.py
  function main (line 9) | def main():

FILE: tgb/datasets/dataset_scripts/tgbl-wiki_neg_generator.py
  function main (line 9) | def main():

FILE: tgb/datasets/dataset_scripts/tgbn-genre.py
  function filter_genre_edgelist (line 55) | def filter_genre_edgelist(fname, genres_dict):
  function get_genre_list (line 80) | def get_genre_list(fname):
  function find_unique_genres (line 133) | def find_unique_genres(fname: str, threshold: float = 0.8):
  function load_genre_dict (line 162) | def load_genre_dict(
  function generate_daily_node_labels (line 180) | def generate_daily_node_labels(fname: str):
  function generate_aggregate_labels (line 246) | def generate_aggregate_labels(fname: str, days: int = 7):
  function most_frequent (line 309) | def most_frequent(List):
  function convert_ts_unix (line 325) | def convert_ts_unix(fname: str, outname: str):
  function convert_ts_edgelist (line 352) | def convert_ts_edgelist(fname: str, outname: str):
  function sort_node_labels (line 378) | def sort_node_labels(fname, outname):
  function sort_edgelist (line 416) | def sort_edgelist(fname, outname="sorted_lastfm_edgelist.csv"):

FILE: tgb/datasets/dataset_scripts/tgbn-reddit.py
  function find_filenames (line 7) | def find_filenames(path_to_dir):
  function combine_edgelist_edgefeat2subreddits (line 17) | def combine_edgelist_edgefeat2subreddits(edgefname, featfname, outname):
  function filter_subreddits (line 63) | def filter_subreddits(fname):
  function clean_edgelist (line 91) | def clean_edgelist(fname, node_counts, outname, threshold=1000):
  function clean_edgelist_reddits (line 123) | def clean_edgelist_reddits(fname, reddit_counts, outname, threshold=50):
  function remove_missing_user (line 155) | def remove_missing_user(fname, outname):
  function generate_daily_node_labels (line 181) | def generate_daily_node_labels(
  function generate_aggregate_labels (line 229) | def generate_aggregate_labels(fname: str, outname: str, days: int = 7):
  function main (line 288) | def main():

FILE: tgb/datasets/dataset_scripts/tgbn-token.py
  function count_node_freq (line 4) | def count_node_freq(fname, filter_size=100):
  function filter_edgelist (line 66) | def filter_edgelist(token_fname, edgefile, outname):
  function filter_by_node (line 104) | def filter_by_node(node_dict, edgefile, outname):
  function store_node_list (line 125) | def store_node_list(node_dict, outname):
  function load_node_dict (line 139) | def load_node_dict(fname):
  function store_token_address (line 161) | def store_token_address(token_dict, outname, topk=1000):
  function analyze_token_frequency (line 180) | def analyze_token_frequency(fname):
  function to_bipartite (line 234) | def to_bipartite(in_name, out_name, node_dict):
  function analyze_csv (line 260) | def analyze_csv(fname):
  function convert_2_sec (line 326) | def convert_2_sec(fname, outname):
  function print_csv (line 357) | def print_csv(fname):
  function sort_edgelist_by_time (line 368) | def sort_edgelist_by_time(fname, outname):
  function generate_aggregate_labels (line 396) | def generate_aggregate_labels(fname: str, outname: str, days: int = 7):
  function main (line 461) | def main():

FILE: tgb/datasets/dataset_scripts/tgbn-trade.py
  function count_unique_countries (line 9) | def count_unique_countries(fname):
  function normalize_edgelist (line 34) | def normalize_edgelist(fname: str, outname: str):
  function generate_aggregate_labels (line 94) | def generate_aggregate_labels(fname: str, outname: str):
  function check_sum_to_one (line 124) | def check_sum_to_one(fname: str):
  function main (line 159) | def main():

FILE: tgb/datasets/tgbl_enron/tgbl-enron_neg_generator.py
  function main (line 7) | def main():

FILE: tgb/datasets/tgbl_lastfm/tgbl-lastfm_neg_generator.py
  function main (line 7) | def main():

FILE: tgb/datasets/tgbl_subreddit/tgbl-subreddit_neg_generator.py
  function main (line 7) | def main():

FILE: tgb/datasets/tgbl_uci/tgbl-uci_neg_generator.py
  function main (line 7) | def main():

FILE: tgb/datasets/thgl_forum/merge_files.py
  function find_filenames (line 7) | def find_filenames(path_to_dir):
  function read_edgelist (line 17) | def read_edgelist(fname, outfname, write_header=False):
  function read_nodeattr (line 47) | def read_nodeattr(fname, outfname, write_header=False):
  function combine_edgelist_edgefeat (line 117) | def combine_edgelist_edgefeat(edgefname, featfname, outname):
  function main (line 192) | def main():

FILE: tgb/datasets/thgl_forum/thgl-forum.py
  function load_csv_raw (line 106) | def load_csv_raw(fname):
  function load_csv_filtered_node (line 199) | def load_csv_filtered_node(fname, low_deg_dict):
  function writeNodeType (line 264) | def writeNodeType(node_type_dict, outname):
  function write2csv (line 275) | def write2csv(outname, out_dict):
  function writeNodeIDMapping (line 292) | def writeNodeIDMapping(node_dict, outname):
  function node_deg_filter (line 302) | def node_deg_filter(node_deg_dict):
  function find_low_degree_nodes (line 321) | def find_low_degree_nodes(node_deg_dict, threshold=10):
  function main (line 332) | def main():

FILE: tgb/datasets/thgl_forum/thgl_forum_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/thgl_github/2024_01/github_extract.py
  function str_to_timestamp (line 47) | def str_to_timestamp(time_str):
  function parse_issue_comment_events (line 52) | def parse_issue_comment_events(event):
  function parse_issue_event (line 80) | def parse_issue_event(event):
  function parse_pull_request_event (line 112) | def parse_pull_request_event(event):
  function parse_pull_request_review_comment_event (line 143) | def parse_pull_request_review_comment_event(event):
  function parse_fork_event (line 169) | def parse_fork_event(event):
  function parse_member_event (line 186) | def parse_member_event(event):
  function parse_event (line 213) | def parse_event(event):
  function parse_file (line 224) | def parse_file(filename):
  function write2csv (line 253) | def write2csv(outname, out_dict):
  function main (line 270) | def main():

FILE: tgb/datasets/thgl_github/2024_02/github_extract.py
  function str_to_timestamp (line 47) | def str_to_timestamp(time_str):
  function parse_issue_comment_events (line 52) | def parse_issue_comment_events(event):
  function parse_issue_event (line 80) | def parse_issue_event(event):
  function parse_pull_request_event (line 112) | def parse_pull_request_event(event):
  function parse_pull_request_review_comment_event (line 143) | def parse_pull_request_review_comment_event(event):
  function parse_fork_event (line 169) | def parse_fork_event(event):
  function parse_member_event (line 186) | def parse_member_event(event):
  function parse_event (line 213) | def parse_event(event):
  function parse_file (line 224) | def parse_file(filename):
  function write2csv (line 253) | def write2csv(outname, out_dict):
  function main (line 270) | def main():

FILE: tgb/datasets/thgl_github/2024_03/github_extract.py
  function str_to_timestamp (line 47) | def str_to_timestamp(time_str):
  function parse_issue_comment_events (line 52) | def parse_issue_comment_events(event):
  function parse_issue_event (line 80) | def parse_issue_event(event):
  function parse_pull_request_event (line 112) | def parse_pull_request_event(event):
  function parse_pull_request_review_comment_event (line 143) | def parse_pull_request_review_comment_event(event):
  function parse_fork_event (line 169) | def parse_fork_event(event):
  function parse_member_event (line 186) | def parse_member_event(event):
  function parse_event (line 213) | def parse_event(event):
  function parse_file (line 224) | def parse_file(filename):
  function write2csv (line 253) | def write2csv(outname, out_dict):
  function main (line 270) | def main():

FILE: tgb/datasets/thgl_github/extract_subset.py
  function load_edgelist (line 4) | def load_edgelist(file_path, freq_threshold=5):
  function subset_by_node (line 88) | def subset_by_node(file_path, low_freq_dict):
  function subset_by_node_type (line 116) | def subset_by_node_type(file_path, remove_node_type_dict, low_freq_dict=...
  function write2csv (line 164) | def write2csv(outname, out_dict):
  function combine_edgelist (line 186) | def combine_edgelist(file_paths, outname):
  function main (line 207) | def main():

FILE: tgb/datasets/thgl_github/thgl_github.py
  function load_csv_raw (line 5) | def load_csv_raw(fname):
  function write2csv (line 31) | def write2csv(outname, out_dict):
  function load_edgelist (line 47) | def load_edgelist(fname):
  function writeNodeType (line 98) | def writeNodeType(node_type_dict, outname):
  function writeEdgeTypeMapping (line 109) | def writeEdgeTypeMapping(edge_type_dict, outname):
  function writeNodeTypeMapping (line 120) | def writeNodeTypeMapping(node_type_dict, outname):
  function write2edgelist (line 131) | def write2edgelist(out_dict, outname):
  function main (line 153) | def main():

FILE: tgb/datasets/thgl_github/thgl_github_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/thgl_myket/thgl_myket.py
  function date2ts (line 11) | def date2ts(date_str: str) -> float:
  function read_csv2dict (line 38) | def read_csv2dict(fname):
  function edge2nodetype (line 83) | def edge2nodetype(out_dict):
  function writeNodeType (line 110) | def writeNodeType(node_type_dict, outname):
  function write2edgelist (line 122) | def write2edgelist(out_dict, outname):
  function main (line 149) | def main():

FILE: tgb/datasets/thgl_myket/thgl_myket_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/thgl_software/thgl_software.py
  function load_csv_raw (line 5) | def load_csv_raw(fname):
  function write2csv (line 31) | def write2csv(outname, out_dict):
  function load_edgelist (line 47) | def load_edgelist(fname):
  function writeNodeType (line 98) | def writeNodeType(node_type_dict, outname):
  function writeEdgeTypeMapping (line 109) | def writeEdgeTypeMapping(edge_type_dict, outname):
  function writeNodeTypeMapping (line 120) | def writeNodeTypeMapping(node_type_dict, outname):
  function write2edgelist (line 131) | def write2edgelist(out_dict, outname):
  function main (line 153) | def main():

FILE: tgb/datasets/thgl_software/thgl_software_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/tkgl_icews/tkgl_icews.py
  function load_csv_raw (line 6) | def load_csv_raw(fname):
  function write2csv (line 47) | def write2csv(outname, out_dict):
  function writeEdgeTypeMapping (line 77) | def writeEdgeTypeMapping(edge_type_dict, outname):
  function main (line 90) | def main():

FILE: tgb/datasets/tkgl_icews/tkgl_icews_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/tkgl_polecat/tkgl_polecat.py
  function load_csv_raw (line 8) | def load_csv_raw(fname):
  function write2csv (line 51) | def write2csv(outname: str,
  function writeEdgeTypeMapping (line 121) | def writeEdgeTypeMapping(edge_type_dict, outname):
  function main (line 132) | def main():

FILE: tgb/datasets/tkgl_polecat/tkgl_polecat_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/tkgl_smallpedia/smallpedia_remove_conflict.py
  function load_static_edgelist (line 3) | def load_static_edgelist(file_path):
  function load_temporal_edgelist (line 24) | def load_temporal_edgelist(file_path):
  function remove_conflict (line 57) | def remove_conflict(static_dict, temporal_dict):
  function write2csv (line 77) | def write2csv(outname: str,
  function main (line 96) | def main():

FILE: tgb/datasets/tkgl_smallpedia/tkgl_smallpedia_ns_gen.py
  function main (line 6) | def main():

FILE: tgb/datasets/tkgl_wikidata/time_edges/tkgl-wikidata_extract.py
  function load_time_csv_raw (line 6) | def load_time_csv_raw(fname):
  function write2csv (line 59) | def write2csv(outname: str,
  function update_dict (line 79) | def update_dict(total_dict, new_dict):
  function retrieve_all_entities (line 95) | def retrieve_all_entities(total_dict):
  function writenode2csv (line 118) | def writenode2csv(outname: str,
  function main (line 131) | def main():

FILE: tgb/datasets/tkgl_wikidata/tkgl-wikidata.py
  function load_time_csv (line 6) | def load_time_csv(fname):
  function write2csv (line 98) | def write2csv(outname: str,
  function extract_subset (line 117) | def extract_subset(fname, outname, start_year=2000, end_year=2024):
  function extract_subset_nodeid (line 152) | def extract_subset_nodeid(fname, outname, start_year=2000, end_year=2024...
  function extract_static_subset (line 193) | def extract_static_subset(fname, outname, node_dict, max_id=1000000):
  function subset_static_edges (line 239) | def subset_static_edges(fname, outname, rel_type, topk=10):
  function main (line 280) | def main():

FILE: tgb/datasets/tkgl_wikidata/tkgl_wikidata_mining.py
  function timeEdgeWrite2csv (line 19) | def timeEdgeWrite2csv(outname, out_dict):
  function EdgeWrite2csv (line 33) | def EdgeWrite2csv(outname, out_dict):
  function main (line 45) | def main():

FILE: tgb/datasets/tkgl_wikidata/tkgl_wikidata_ns_gen.py
  function main (line 12) | def main():

FILE: tgb/datasets/tkgl_wikidata/wikidata_remove_conflict.py
  function load_static_edgelist (line 3) | def load_static_edgelist(file_path):
  function load_temporal_edgelist (line 24) | def load_temporal_edgelist(file_path):
  function remove_conflict (line 57) | def remove_conflict(static_dict, temporal_dict):
  function write2csv (line 77) | def write2csv(outname: str,
  function main (line 96) | def main():

FILE: tgb/datasets/tkgl_yago/tkgl_yago.py
  function main (line 7) | def main():
  function write_csv (line 30) | def write_csv(outname, out_dict):
  function load_csv (line 43) | def load_csv(fname):

FILE: tgb/datasets/tkgl_yago/tkgl_yago_ns_gen.py
  function main (line 8) | def main():

FILE: tgb/linkproppred/dataset.py
  class LinkPropPredDataset (line 42) | class LinkPropPredDataset(object):
    method __init__ (line 43) | def __init__(
    method _version_check (line 176) | def _version_check(self) -> None:
    method download (line 201) | def download(self) -> None:
    method generate_processed_files (line 244) | def generate_processed_files(self) -> pd.DataFrame:
    method pre_process (line 347) | def pre_process(self):
    method generate_splits (line 400) | def generate_splits(
    method preprocess_static_edges (line 430) | def preprocess_static_edges(self):
    method eval_metric (line 450) | def eval_metric(self) -> str:
    method negative_sampler (line 459) | def negative_sampler(self) -> NegativeEdgeSampler:
    method load_val_ns (line 468) | def load_val_ns(self) -> None:
    method load_test_ns (line 476) | def load_test_ns(self) -> None:
    method num_nodes (line 485) | def num_nodes(self) -> int:
    method num_edges (line 499) | def num_edges(self) -> int:
    method num_rels (line 510) | def num_rels(self) -> int:
    method node_feat (line 523) | def node_feat(self) -> Optional[np.ndarray]:
    method node_type (line 532) | def node_type(self) -> Optional[np.ndarray]:
    method edge_feat (line 541) | def edge_feat(self) -> Optional[np.ndarray]:
    method edge_type (line 550) | def edge_type(self) -> Optional[np.ndarray]:
    method static_data (line 559) | def static_data(self) -> Optional[np.ndarray]:
    method full_data (line 570) | def full_data(self) -> Dict[str, Any]:
    method train_mask (line 584) | def train_mask(self) -> np.ndarray:
    method val_mask (line 595) | def val_mask(self) -> np.ndarray:
    method test_mask (line 606) | def test_mask(self) -> np.ndarray:
  function main (line 617) | def main():

FILE: tgb/linkproppred/dataset_pyg.py
  class PyGLinkPropPredDataset (line 9) | class PyGLinkPropPredDataset(Dataset):
    method __init__ (line 10) | def __init__(
    method eval_metric (line 54) | def eval_metric(self) -> str:
    method negative_sampler (line 63) | def negative_sampler(self) -> NegativeEdgeSampler:
    method num_nodes (line 72) | def num_nodes(self) -> int:
    method num_rels (line 81) | def num_rels(self) -> int:
    method num_edges (line 90) | def num_edges(self) -> int:
    method load_val_ns (line 98) | def load_val_ns(self) -> None:
    method load_test_ns (line 104) | def load_test_ns(self) -> None:
    method train_mask (line 111) | def train_mask(self) -> torch.Tensor:
    method val_mask (line 122) | def val_mask(self) -> torch.Tensor:
    method test_mask (line 133) | def test_mask(self) -> torch.Tensor:
    method node_feat (line 144) | def node_feat(self) -> torch.Tensor:
    method node_type (line 153) | def node_type(self) -> torch.Tensor:
    method src (line 162) | def src(self) -> torch.Tensor:
    method dst (line 171) | def dst(self) -> torch.Tensor:
    method ts (line 180) | def ts(self) -> torch.Tensor:
    method static_data (line 189) | def static_data(self) -> torch.Tensor:
    method edge_type (line 206) | def edge_type(self) -> torch.Tensor:
    method edge_feat (line 215) | def edge_feat(self) -> torch.Tensor:
    method edge_label (line 224) | def edge_label(self) -> torch.Tensor:
    method process_data (line 232) | def process_data(self) -> None:
    method get_TemporalData (line 286) | def get_TemporalData(self) -> TemporalData:
    method len (line 311) | def len(self) -> int:
    method get (line 319) | def get(self, idx: int) -> TemporalData:
    method __repr__ (line 346) | def __repr__(self) -> str:

FILE: tgb/linkproppred/evaluate.py
  class Evaluator (line 16) | class Evaluator(object):
    method __init__ (line 19) | def __init__(self, name: str, k_value: int = 10):
    method _parse_and_check_input (line 31) | def _parse_and_check_input(self, input_dict):
    method _eval_hits_and_mrr (line 72) | def _eval_hits_and_mrr(self, y_pred_pos, y_pred_neg, type_info, k_value):
    method eval (line 118) | def eval(self,

FILE: tgb/linkproppred/negative_generator.py
  class NegativeEdgeGenerator (line 16) | class NegativeEdgeGenerator(object):
    method __init__ (line 17) | def __init__(
    method generate_negative_samples (line 69) | def generate_negative_samples(self,
    method generate_negative_samples_rnd (line 103) | def generate_negative_samples_rnd(self,
    method generate_historical_edge_set (line 172) | def generate_historical_edge_set(self,
    method generate_negative_samples_hist_rnd (line 208) | def generate_negative_samples_hist_rnd(

FILE: tgb/linkproppred/negative_sampler.py
  class NegativeEdgeSampler (line 15) | class NegativeEdgeSampler(object):
    method __init__ (line 16) | def __init__(
    method load_eval_set (line 45) | def load_eval_set(
    method reset_eval_set (line 67) | def reset_eval_set(self,
    method query_batch (line 85) | def query_batch(self,

FILE: tgb/linkproppred/thg_negative_generator.py
  class THGNegativeEdgeGenerator (line 21) | class THGNegativeEdgeGenerator(object):
    method __init__ (line 22) | def __init__(
    method get_destinations_based_on_node_type (line 70) | def get_destinations_based_on_node_type(self,
    method generate_negative_samples (line 101) | def generate_negative_samples(self,
    method generate_negative_samples_nt (line 133) | def generate_negative_samples_nt(self,
    method generate_negative_samples_random (line 212) | def generate_negative_samples_random(self,

FILE: tgb/linkproppred/thg_negative_sampler.py
  class THGNegativeEdgeSampler (line 14) | class THGNegativeEdgeSampler(object):
    method __init__ (line 15) | def __init__(
    method load_eval_set (line 45) | def load_eval_set(
    method query_batch (line 67) | def query_batch(self,

FILE: tgb/linkproppred/tkg_negative_generator.py
  class TKGNegativeEdgeGenerator (line 20) | class TKGNegativeEdgeGenerator(object):
    method __init__ (line 21) | def __init__(
    method generate_dst_dict (line 77) | def generate_dst_dict(self, edge_data: TemporalData, dst_name: str) ->...
    method generate_negative_samples (line 126) | def generate_negative_samples(self,
    method generate_negative_samples_ftr (line 160) | def generate_negative_samples_ftr(self,
    method generate_negative_samples_dst (line 224) | def generate_negative_samples_dst(self,
    method generate_negative_samples_random (line 328) | def generate_negative_samples_random(self,

FILE: tgb/linkproppred/tkg_negative_sampler.py
  class TKGNegativeEdgeSampler (line 17) | class TKGNegativeEdgeSampler(object):
    method __init__ (line 18) | def __init__(
    method load_eval_set (line 48) | def load_eval_set(
    method query_batch (line 70) | def query_batch(self,

FILE: tgb/nodeproppred/dataset.py
  class NodePropPredDataset (line 29) | class NodePropPredDataset(object):
    method __init__ (line 30) | def __init__(
    method _version_check (line 112) | def _version_check(self) -> None:
    method download (line 132) | def download(self) -> None:
    method generate_processed_files (line 178) | def generate_processed_files(
    method pre_process (line 256) | def pre_process(self) -> None:
    method generate_splits (line 297) | def generate_splits(
    method find_next_labels_batch (line 327) | def find_next_labels_batch(
    method reset_label_time (line 360) | def reset_label_time(self) -> None:
    method return_label_ts (line 368) | def return_label_ts(self) -> int:
    method num_classes (line 380) | def num_classes(self) -> int:
    method eval_metric (line 389) | def eval_metric(self) -> str:
    method node_feat (line 399) | def node_feat(self) -> Optional[np.ndarray]:
    method edge_feat (line 409) | def edge_feat(self) -> Optional[np.ndarray]:
    method node_label_dict (line 418) | def node_label_dict(self) -> Dict[int, Dict[int, Any]]:
    method full_data (line 427) | def full_data(self) -> Dict[str, Any]:
    method train_mask (line 441) | def train_mask(self) -> np.ndarray:
    method val_mask (line 452) | def val_mask(self) -> np.ndarray:
    method test_mask (line 464) | def test_mask(self) -> np.ndarray:
  function main (line 476) | def main():

FILE: tgb/nodeproppred/dataset_pyg.py
  class PyGNodePropPredDataset (line 14) | class PyGNodePropPredDataset(InMemoryDataset):
    method __init__ (line 28) | def __init__(
    method num_classes (line 48) | def num_classes(self) -> int:
    method eval_metric (line 57) | def eval_metric(self) -> str:
    method train_mask (line 66) | def train_mask(self) -> torch.Tensor:
    method val_mask (line 77) | def val_mask(self) -> torch.Tensor:
    method test_mask (line 88) | def test_mask(self) -> torch.Tensor:
    method src (line 99) | def src(self) -> torch.Tensor:
    method dst (line 108) | def dst(self) -> torch.Tensor:
    method ts (line 117) | def ts(self) -> torch.Tensor:
    method edge_feat (line 126) | def edge_feat(self) -> torch.Tensor:
    method edge_label (line 135) | def edge_label(self) -> torch.Tensor:
    method process_data (line 143) | def process_data(self):
    method get_TemporalData (line 174) | def get_TemporalData(
    method reset_label_time (line 191) | def reset_label_time(self) -> None:
    method get_node_label (line 197) | def get_node_label(self, cur_t):
    method get_label_time (line 210) | def get_label_time(self) -> int:
    method len (line 218) | def len(self) -> int:
    method get (line 226) | def get(self, idx: int) -> TemporalData:
    method __repr__ (line 243) | def __repr__(self) -> str:

FILE: tgb/nodeproppred/evaluate.py
  class Evaluator (line 15) | class Evaluator(object):
    method __init__ (line 18) | def __init__(self, name: str):
    method _parse_and_check_input (line 28) | def _parse_and_check_input(self, input_dict):
    method _compute_metrics (line 75) | def _compute_metrics(self, y_true, y_pred):
    method eval (line 94) | def eval(self, input_dict, verbose=False):
    method expected_input_format (line 108) | def expected_input_format(self):
    method expected_output_format (line 119) | def expected_output_format(self):
  function main (line 129) | def main():

FILE: tgb/utils/dataset_stats.py
  function get_unique_edges (line 17) | def get_unique_edges(sources, destination):
  function get_avg_e_per_ts (line 28) | def get_avg_e_per_ts(edgelist_df):
  function get_avg_degree (line 41) | def get_avg_degree(edgelist_df):
  function get_index_metrics (line 58) | def get_index_metrics(train_val_data, test_data):
  function get_node_ratio (line 87) | def get_node_ratio(history_data, eval_data):
  function get_dataset_stats (line 103) | def get_dataset_stats(data, temporal_stats=False):
  function main (line 148) | def main():

FILE: tgb/utils/info.py
  class BColors (line 10) | class BColors:

FILE: tgb/utils/pre_process.py
  function process_node_type (line 16) | def process_node_type(
  function csv_to_forum_data (line 49) | def csv_to_forum_data(
  function csv_to_thg_data (line 130) | def csv_to_thg_data(
  function csv_to_wikidata (line 201) | def csv_to_wikidata(
  function csv_to_staticdata (line 275) | def csv_to_staticdata(
  function csv_to_tkg_data (line 333) | def csv_to_tkg_data(
  function load_edgelist_wiki (line 408) | def load_edgelist_wiki(fname: str) -> pd.DataFrame:
  function load_edgelist_trade (line 437) | def load_edgelist_trade(fname: str, label_size=255):
  function load_trade_label_dict (line 491) | def load_trade_label_dict(
  function load_edgelist_token (line 540) | def load_edgelist_token(
  function load_edgelist_sr (line 639) | def load_edgelist_sr(
  function load_labels_sr (line 715) | def load_labels_sr(
  function load_label_dict (line 772) | def load_label_dict(fname: str, node_ids: dict, rd_dict: dict) -> dict:
  function csv_to_pd_data_rc (line 817) | def csv_to_pd_data_rc(
  function csv_to_pd_data_sc (line 896) | def csv_to_pd_data_sc(
  function convert_str2int (line 982) | def convert_str2int(
  function csv_to_pd_data (line 1006) | def csv_to_pd_data(
  function process_node_feat (line 1114) | def process_node_feat(
  function clean_rows (line 1200) | def clean_rows(
  function load_edgelist_datetime (line 1244) | def load_edgelist_datetime(fname, label_size=514):
  function load_genre_list (line 1309) | def load_genre_list(fname):
  function reindex (line 1338) | def reindex(

FILE: tgb/utils/stats.py
  function analyze_csv (line 15) | def analyze_csv(fname):
  function plot_curve (line 80) | def plot_curve(y: np.ndarray, outname: str) -> None:
  function main (line 92) | def main():

FILE: tgb/utils/utils.py
  function set_verbose (line 17) | def set_verbose(flag: bool) -> None:
  function vprint (line 21) | def vprint(*args, **kwargs):
  function add_inverse_quadruples (line 27) | def add_inverse_quadruples(df: pd.DataFrame) -> pd.DataFrame:
  function add_inverse_quadruples_np (line 65) | def add_inverse_quadruples_np(quadruples: np.array,
  function add_inverse_quadruples_pyg (line 80) | def add_inverse_quadruples_pyg(data: TemporalData, num_rels:int=-1) -> l...
  function save_pkl (line 102) | def save_pkl(obj: Any, fname: str) -> None:
  function load_pkl (line 110) | def load_pkl(fname: str) -> Any:
  function set_random_seed (line 117) | def set_random_seed(random_seed: int):
  function find_nearest (line 134) | def find_nearest(array, value):
  function get_args (line 140) | def get_args():
  function save_results (line 165) | def save_results(new_results: dict, filename: str):
  function split_by_time (line 187) | def split_by_time(data):