MCPcopy Create free account

hub / github.com/DeepGraphLearning/FewShotRE / functions

Functions61 in github.com/DeepGraphLearning/FewShotRE

↓ 5 callersMethodtokenize
(self, raw_tokens, pos_head, pos_tail)
fewshot_re_kit/sentence_encoder.py:29
↓ 4 callersMethoditem
PyTorch before and after 0.4
fewshot_re_kit/framework.py:90
↓ 3 callersFunctionget_loader
(name, encoder, N, K, Q, batch_size, num_workers=8, collate_fn=collate_fn, na_rate=0, root='./data')
fewshot_re_kit/data_loader.py:110
↓ 3 callersFunctionget_loader_pair
(name, encoder, N, K, Q, batch_size, num_workers=8, collate_fn=collate_fn_pair, na_rate=0, root='./da
fewshot_re_kit/data_loader.py:224
↓ 2 callersMethod__additem__
(self, d, word, pos, mask, rel)
fewshot_re_kit/data_loader.py:32
↓ 2 callersMethod__get_emb__
(self, embs, pos)
models/regrab.py:51
↓ 2 callersMethod__getraw__
(self, item)
fewshot_re_kit/data_loader.py:140
↓ 2 callersMethod__init__
(self, word_vec_mat, word2id, max_length, word_embedding_dim=50, pos_embedding_dim=5, hidden_size
fewshot_re_kit/sentence_encoder.py:13
↓ 2 callersMethod__load_model__
ckpt: Path of the checkpoint return: Checkpoint dict
fewshot_re_kit/framework.py:78
↓ 2 callersMethodaccuracy
pred: Prediction results with whatever size label: Label with whatever size return: [Accuracy] (A single value)
fewshot_re_kit/framework.py:52
↓ 2 callersMethodeval
model: a FewShotREModel instance B: Batch size N: Num of classes for each batch K: Num of instances for each class in
fewshot_re_kit/framework.py:295
↓ 1 callersMethod__additem__
(self, d, word, pos1, pos2, mask)
fewshot_re_kit/data_loader.py:258
↓ 1 callersMethod__dist__
(self, x, y, dim)
models/regrab.py:45
↓ 1 callersMethod__getraw__
(self, item)
fewshot_re_kit/data_loader.py:26
↓ 1 callersMethod__getraw__
(self, item)
fewshot_re_kit/data_loader.py:252
↓ 1 callersMethod_load_preprocessed_file
(self)
fewshot_re_kit/old_data_loader.py:21
↓ 1 callersMethodcnn
(self, inputs)
fewshot_re_kit/network/encoder.py:27
↓ 1 callersFunctionget_loader_unsupervised
(name, encoder, N, K, Q, batch_size, num_workers=8, collate_fn=collate_fn_unsupervised, na_rate=0, ro
fewshot_re_kit/data_loader.py:293
↓ 1 callersMethodloss
logits: Logits with the size (..., class_num) label: Label with whatever size. return: [Loss] (A single value)
fewshot_re_kit/framework.py:43
↓ 1 callersFunctionmain
()
train_demo.py:17
↓ 1 callersMethodnext_one
(self, N, K, Q)
fewshot_re_kit/old_data_loader.py:210
↓ 1 callersMethodset_reladj
(self, reladj)
models/regrab.py:42
↓ 1 callersMethodset_relemb
(self, rel2id, relemb)
models/regrab.py:38
↓ 1 callersMethodtrain
model: a FewShotREModel instance model_name: Name of the model B: Batch size N: Num of classes for each batch
fewshot_re_kit/framework.py:100
Method__additem__
(self, d, word, pos1, pos2, mask)
fewshot_re_kit/data_loader.py:146
Method__batch_dist__
(self, S, Q)
models/regrab.py:48
Method__getitem__
(self, index)
fewshot_re_kit/data_loader.py:38
Method__getitem__
(self, index)
fewshot_re_kit/data_loader.py:152
Method__getitem__
(self, index)
fewshot_re_kit/data_loader.py:264
Method__init__
sentence_encoder: Sentence encoder You need to set self.cost as your own loss function.
fewshot_re_kit/framework.py:22
Method__init__
train_data_loader: DataLoader for training. val_data_loader: DataLoader for validating. test_data_loader: DataLoader for test
fewshot_re_kit/framework.py:62
Method__init__
(self, name, encoder, N, K, Q, na_rate, root)
fewshot_re_kit/data_loader.py:12
Method__init__
(self, name, encoder, N, K, Q, na_rate, root)
fewshot_re_kit/data_loader.py:125
Method__init__
(self, name, encoder, N, K, Q, na_rate, root)
fewshot_re_kit/data_loader.py:239
Method__init__
(self, pretrain_path, max_length)
fewshot_re_kit/sentence_encoder.py:62
Method__init__
(self, pretrain_path, max_length)
fewshot_re_kit/sentence_encoder.py:125
Method__init__
file_name: Json file storing the data in the following format { "P155": # relation id [
fewshot_re_kit/old_data_loader.py:59
Method__init__
(self, max_length, word_embedding_dim=50, pos_embedding_dim=5, hidden_size=230)
fewshot_re_kit/network/encoder.py:9
Method__init__
(self, word_vec_mat, max_length, word_embedding_dim=50, pos_embedding_dim=5)
fewshot_re_kit/network/embedding.py:9
Method__init__
(self, hidden_size=230, num_labels=2)
models/d.py:11
Method__init__
(self, sentence_encoder, hidden_size=230, eps=0.1, temp=100.0, step=10, smp=1, ratio=1.0, wtp=1.0, wtn=1.0, wt
models/regrab.py:12
Method__len__
(self)
fewshot_re_kit/data_loader.py:85
Method__len__
(self)
fewshot_re_kit/data_loader.py:208
Method__len__
(self)
fewshot_re_kit/data_loader.py:280
Functioncollate_fn
(data)
fewshot_re_kit/data_loader.py:88
Functioncollate_fn_pair
(data)
fewshot_re_kit/data_loader.py:211
Functioncollate_fn_unsupervised
(data)
fewshot_re_kit/data_loader.py:283
Methodforward
support: Inputs of the support set. query: Inputs of the query set. N: Num of classes K: Num of instances for each cl
fewshot_re_kit/framework.py:32
Methodforward
(self, inputs)
fewshot_re_kit/sentence_encoder.py:24
Methodforward
(self, inputs)
fewshot_re_kit/sentence_encoder.py:68
Methodforward
(self, inputs)
fewshot_re_kit/sentence_encoder.py:137
Methodforward
(self, inputs)
fewshot_re_kit/network/encoder.py:24
Methodforward
(self, inputs)
fewshot_re_kit/network/embedding.py:27
Methodforward
(self, x)
models/d.py:20
Methodforward
support: Inputs of the support set. query: Inputs of the query set. N: Num of classes K: Num of instances for each cl
models/regrab.py:57
Methodnext_batch
B: batch size. N: the number of relations for each batch K: the number of support instances for each relation Q: the
fewshot_re_kit/old_data_loader.py:10
Methodnext_batch
(self, B, N, K, Q)
fewshot_re_kit/old_data_loader.py:256
Methodpcnn
(self, inputs, mask)
fewshot_re_kit/network/encoder.py:33
Methodtokenize
(self, raw_tokens, pos_head, pos_tail)
fewshot_re_kit/sentence_encoder.py:72
Methodtokenize
(self, raw_tokens, pos_head, pos_tail)
fewshot_re_kit/sentence_encoder.py:144
Functionwarmup_linear
(global_step, warmup_step)
fewshot_re_kit/framework.py:15