SYMBOL INDEX (78 symbols across 7 files)

FILE: nbfnet/dataset.py
  class InductiveKnowledgeGraphDataset (line 14) | class InductiveKnowledgeGraphDataset(data.KnowledgeGraphDataset):
    method load_inductive_tsvs (line 16) | def load_inductive_tsvs(self, train_files, test_files, verbose=0):
    method __getitem__ (line 86) | def __getitem__(self, index):
    method split (line 89) | def split(self):
  class CoraLinkPrediction (line 100) | class CoraLinkPrediction(datasets.Cora):
    method __init__ (line 102) | def __init__(self, **kwargs):
    method __getitem__ (line 106) | def __getitem__(self, index):
    method __len__ (line 109) | def __len__(self):
    method split (line 112) | def split(self, ratios=(85, 5, 10)):
  class CiteSeerLinkPrediction (line 124) | class CiteSeerLinkPrediction(datasets.CiteSeer):
    method __init__ (line 126) | def __init__(self, **kwargs):
    method __getitem__ (line 130) | def __getitem__(self, index):
    method __len__ (line 133) | def __len__(self):
    method split (line 136) | def split(self, ratios=(85, 5, 10)):
  class PubMedLinkPrediction (line 148) | class PubMedLinkPrediction(datasets.PubMed):
    method __init__ (line 150) | def __init__(self, **kwargs):
    method __getitem__ (line 154) | def __getitem__(self, index):
    method __len__ (line 157) | def __len__(self):
    method split (line 160) | def split(self, ratios=(85, 5, 10)):
  class FB15k237Inductive (line 172) | class FB15k237Inductive(InductiveKnowledgeGraphDataset):
    method __init__ (line 184) | def __init__(self, path, version="v1", verbose=1):
  class WN18RRInductive (line 211) | class WN18RRInductive(InductiveKnowledgeGraphDataset):
    method __init__ (line 223) | def __init__(self, path, version="v1", verbose=1):
  class OGBLBioKG (line 250) | class OGBLBioKG(data.KnowledgeGraphDataset):
    method __init__ (line 252) | def __init__(self, path, verbose=1):
    method load_ogb (line 259) | def load_ogb(self, dataset, verbose=1):
    method split (line 333) | def split(self, test_negative=True):

FILE: nbfnet/layer.py
  class GeneralizedRelationalConv (line 11) | class GeneralizedRelationalConv(layers.MessagePassingBase):
    method __init__ (line 20) | def __init__(self, input_dim, output_dim, num_relation, query_input_di...
    method message (line 49) | def message(self, graph, input):
    method aggregate (line 78) | def aggregate(self, graph, message):
    method message_and_aggregate (line 108) | def message_and_aggregate(self, graph, input):
    method combine (line 160) | def combine(self, input, update):

FILE: nbfnet/model.py
  class NeuralBellmanFordNetwork (line 17) | class NeuralBellmanFordNetwork(nn.Module, core.Configurable):
    method __init__ (line 19) | def __init__(self, input_dim, hidden_dims, num_relation=None, symmetri...
    method remove_easy_edges (line 51) | def remove_easy_edges(self, graph, h_index, t_index, r_index=None):
    method negative_sample_to_tail (line 70) | def negative_sample_to_tail(self, h_index, t_index, r_index):
    method as_relational_graph (line 79) | def as_relational_graph(self, graph, self_loop=True):
    method bellmanford (line 95) | def bellmanford(self, graph, h_index, r_index, separate_grad=False):
    method forward (line 132) | def forward(self, graph, h_index, t_index, r_index=None, all_loss=None...
    method visualize (line 164) | def visualize(self, graph, h_index, t_index, r_index):
    method beam_search_distance (line 187) | def beam_search_distance(self, graphs, h_index, t_index, num_beam=10):
    method topk_average_length (line 242) | def topk_average_length(self, distances, back_edges, t_index, k=10):

FILE: nbfnet/task.py
  class KnowledgeGraphCompletionExt (line 20) | class KnowledgeGraphCompletionExt(tasks.KnowledgeGraphCompletion, core.C...
    method __init__ (line 22) | def __init__(self, model, criterion="bce",
    method preprocess (line 30) | def preprocess(self, train_set, valid_set, test_set):
    method target (line 54) | def target(self, batch):
    method evaluate (line 60) | def evaluate(self, pred, target):
  class LinkPrediction (line 104) | class LinkPrediction(tasks.Task, core.Configurable):
    method __init__ (line 108) | def __init__(self, model, criterion="bce", metric=("auroc", "ap"), num...
    method preprocess (line 116) | def preprocess(self, train_set, valid_set, test_set):
    method forward (line 130) | def forward(self, batch):
    method _strict_negative (line 154) | def _strict_negative(self, count, split="train"):
    method predict_and_target (line 176) | def predict_and_target(self, batch, all_loss=None, metric=None):
    method evaluate (line 200) | def evaluate(self, pred, target):
  class InductiveKnowledgeGraphCompletion (line 220) | class InductiveKnowledgeGraphCompletion(tasks.KnowledgeGraphCompletion, ...
    method __init__ (line 222) | def __init__(self, model, criterion="bce", metric=("mr", "mrr", "hits@...
    method preprocess (line 229) | def preprocess(self, train_set, valid_set, test_set):
    method predict (line 253) | def predict(self, batch, all_loss=None, metric=None):
    method target (line 294) | def target(self, batch):
    method evaluate (line 321) | def evaluate(self, pred, target):
  class KnowledgeGraphCompletionOGB (line 358) | class KnowledgeGraphCompletionOGB(tasks.KnowledgeGraphCompletion, core.C...
    method __init__ (line 360) | def __init__(self, model, criterion="bce", evaluator=None, num_negativ...
    method preprocess (line 369) | def preprocess(self, train_set, valid_set, test_set):
    method _strict_negative (line 399) | def _strict_negative(self, pos_h_index, pos_t_index, pos_r_index):
    method predict (line 438) | def predict(self, batch, all_loss=None, metric=None):
    method target (line 466) | def target(self, batch):
    method evaluate (line 473) | def evaluate(self, pred, target):

FILE: nbfnet/util.py
  function get_root_logger (line 22) | def get_root_logger(file=True):
  function create_working_directory (line 35) | def create_working_directory(cfg):
  function detect_variables (line 62) | def detect_variables(cfg_file):
  function load_config (line 71) | def load_config(cfg_file, context=None):
  function parse_args (line 81) | def parse_args():
  function build_solver (line 98) | def build_solver(cfg, dataset):

FILE: script/run.py
  function train_and_validate (line 15) | def train_and_validate(cfg, solver):
  function test (line 40) | def test(cfg, solver):

FILE: script/visualize.py
  function load_vocab (line 18) | def load_vocab(dataset):
  function visualize_path (line 31) | def visualize_path(solver, triplet, entity_vocab, relation_vocab):