Functions — github.com/atcbosselut/comet-commonsense

hub / github.com/atcbosselut/comet-commonsense / functions

Functions222 in github.com/atcbosselut/comet-commonsense

⨍ Functions 222◇ Types & classes 28

↓ 15 callersMethodload_data

(self, path)

src/data/atomic.py:79

↓ 11 callersMethodreset_offsets

(self, splits=["train", "test", "dev"], shuffle=True, keys=None)

src/data/atomic.py:171

↓ 7 callersMethod__init__

(self, n_state, cfg)

src/models/gpt.py:137

↓ 7 callersMethodsample_batch

(self, split, bs, idxs=None)

src/data/atomic.py:139

↓ 6 callersMethodencode

(self, texts, verbose=True)

src/data/utils.py:114

↓ 5 callersMethodget_tracked_score

(self)

src/train/atomic_train.py:72

↓ 5 callersFunctionmake_name_string

(dict_, final=False, do_epoch=False, set_epoch=None)

utils/utils.py:57

↓ 3 callersMethod__init__

(self, opt, data_loader, batch_mode=False)

src/evaluate/sampler.py:16

↓ 3 callersMethodgenerate_sequence

(self, batch, model)

src/evaluate/sampler.py:22

↓ 3 callersMethodmake_tensors

(self, text_encoder, special, splits=["train", "dev", "test"], test=False)

src/data/atomic.py:104

↓ 3 callersMethodrun

(self)

src/train/train.py:88

↓ 3 callersMethodsplit_heads

(self, x, k=False)

src/models/gpt.py:115

↓ 2 callersMethodbpe

(self, token)

src/data/utils.py:69

↓ 2 callersMethodcounter

(self, nums)

src/evaluate/atomic_evaluate.py:33

↓ 2 callersMethoddo_backward_pass

(self, loss)

src/train/train.py:162

↓ 2 callersMethoddo_forward_pass

(self, nums)

src/train/train.py:156

↓ 2 callersMethodepoch

(self)

src/train/train.py:98

↓ 2 callersFunctiongetVec

(We, words, t)

scripts/classify/demo_bilinear.py:14

↓ 2 callersFunctionget_pairs

Return set of symbol pairs in a word. word is represented as tuple of symbols (symbols being variable-length strings)

src/data/utils.py:25

↓ 2 callersFunctionhandle_underscores

(suffix, text_encoder, prefix=False)

src/data/atomic.py:242

↓ 2 callersMethodinitialize_losses

(self, categories)

src/train/atomic_train.py:42

↓ 2 callersFunctionis_bool

(v)

utils/utils.py:87

↓ 2 callersMethodlog_losses

(self, opt, losses)

src/train/train.py:70

↓ 2 callersFunctionmain

(num)

src/main_conceptnet.py:17

↓ 2 callersFunctionmle_steps

(key, model, input_, targets, attention_mask, loss_reduction="mean", i=None)

src/train/batch.py:105

↓ 2 callersMethodoffset_summary

(self, split)

src/data/atomic.py:45

↓ 2 callersMethodreset_losses

(self)

src/train/train.py:171

↓ 2 callersMethodrun_evaluation_cycle

(self)

src/train/train.py:133

↓ 2 callersMethodsave_model

(self, tracked_score)

src/train/train.py:56

↓ 2 callersMethodset_evaluator

(self, opt, model, data_loader)

src/train/atomic_train.py:19

↓ 2 callersMethodset_logger

(self)

src/train/train.py:76

↓ 2 callersMethodset_sampler

(self, opt)

src/train/atomic_train.py:27

↓ 2 callersMethodstop_logger

(self)

src/train/train.py:85

↓ 2 callersFunctiontext_standardize

fixes some issues the spacy tokenizer had on books corpus also does some whitespace standardization

src/data/utils.py:38

↓ 2 callersFunctionupdate_generation_losses

(losses, nums, micro, macro, bs, length, loss, split)

src/train/batch.py:128

↓ 2 callersMethodupdate_parameters

(self)

src/train/train.py:165

↓ 2 callersMethodupdate_top_score

(self, opt)

src/train/atomic_train.py:62

↓ 1 callersMethod__init__

(self, opt)

src/data/atomic.py:24

↓ 1 callersMethod__init__

(self, opt, meta, data_loader, model, optimizer)

src/train/train.py:20

↓ 1 callersMethod_attn

(self, q, k, v, sequence_mask)

src/models/gpt.py:93

↓ 1 callersFunctionappend_batch

(X, next_idx, mask)

scripts/generate/generate_atomic_topk.py:129

↓ 1 callersFunctionappend_batch

(X, beam_toks, mask)

scripts/generate/generate_atomic_beam_search.py:139

↓ 1 callersFunctionappend_batch

(X, beam_toks, mask)

scripts/generate/generate_conceptnet_beam_search.py:155

↓ 1 callersFunctionappend_batch

(X, next_idx, mask)

scripts/generate/generate_atomic_greedy.py:139

↓ 1 callersMethodappend_batch

(self, X, next_idx, mask)

src/evaluate/sampler.py:30

↓ 1 callersMethodappend_batch

(self, X, next_idx, mask)

src/evaluate/sampler.py:101

↓ 1 callersMethodappend_batch

(self, X, beam_toks, mask)

src/evaluate/sampler.py:199

↓ 1 callersMethodbatch

(self, opt, nums, average_loss, batch_variables, eval_mode)

src/evaluate/conceptnet_evaluate.py:23

↓ 1 callersMethodbatch

(self, opt, *args)

src/train/atomic_train.py:33

↓ 1 callersMethodclip_gradients

(self)

src/train/train.py:151

↓ 1 callersFunctioncompile_final_sequence

(final_event1, final_event2, final_relation, text_encoder)

src/data/conceptnet.py:333

↓ 1 callersFunctioncompile_final_sequence

(opt, final_prefix, final_suffix, category, text_encoder)

src/data/atomic.py:318

↓ 1 callersMethodcompute_final_scores

(self, average_loss, nums)

src/evaluate/atomic_evaluate.py:24

↓ 1 callersMethodcounter

(self, nums)

src/train/atomic_train.py:75

↓ 1 callersMethodcycle

(self, bar, cycle_num)

src/train/train.py:205

↓ 1 callersMethoddecide_to_save

(self)

src/train/train.py:45

↓ 1 callersFunctiondecode

(model, input_, attention_mask, i=None)

src/train/batch.py:124

↓ 1 callersFunctiondo_example

(text_encoder, event1, relation, event2)

src/data/conceptnet.py:318

↓ 1 callersFunctiondo_example

(text_encoder, prefix, suffix, do_prefix, do_suffix)

src/data/atomic.py:300

↓ 1 callersFunctiondo_take_partial_dataset

(data_opts)

src/data/atomic.py:49

↓ 1 callersMethodepoch

(self, opt, model, data_loader, split, keyset=None)

src/evaluate/evaluate.py:35

↓ 1 callersFunctionfind_underscore_length

(seq)

src/data/atomic.py:234

↓ 1 callersMethodgenerate

(self, split="dev")

src/evaluate/generate.py:67

↓ 1 callersMethodgenerate_batch

(self, sequences, split, verbose=False, bs=1)

src/evaluate/conceptnet_generate.py:84

↓ 1 callersFunctionget_data_parameters

(opt, experiment, dataset)

src/data/config.py:85

↓ 1 callersFunctionget_data_params

(gens_file)

scripts/evaluate/bleu_atomic.py:27

↓ 1 callersFunctionget_eval_parameters

(opt, force_categories=None)

src/data/config.py:58

↓ 1 callersFunctionget_event

(event)

scripts/evaluate/bleu_atomic.py:81

↓ 1 callersFunctionget_generation_sequences

(data, split, text_encoder, test, max_e1=10, max_e2=15)

src/data/conceptnet.py:285

↓ 1 callersFunctionget_generation_sequences

(opt, data, split, text_encoder, test)

src/data/atomic.py:265

↓ 1 callersFunctionget_net_parameters

(opt)

src/data/config.py:146

↓ 1 callersMethodget_tracked_score

(self)

src/train/conceptnet_train.py:52

↓ 1 callersFunctionget_training_parameters

(opt)

src/data/config.py:110

↓ 1 callersMethodinitialize_losses

(self)

src/evaluate/atomic_evaluate.py:19

↓ 1 callersFunctionload_atomic_data

(opt)

src/interactive/functions.py:37

↓ 1 callersFunctionload_conceptnet_data

(opt)

src/interactive/functions.py:52

↓ 1 callersFunctionload_openai_pretrained_model

(model, n_ctx=-1, n_special=-1, n_transfer=12, n_embd=768, path='./model/', p

src/models/gpt.py:235

↓ 1 callersFunctionmake_attention_mask

(sequences)

src/data/conceptnet.py:274

↓ 1 callersFunctionmake_attention_mask

(sequences)

src/data/atomic.py:222

↓ 1 callersFunctionmake_loss_mask

(sequences, max_event)

src/data/conceptnet.py:278

↓ 1 callersFunctionmake_loss_mask

(sequences, max_event, num_delim_tokens)

src/data/atomic.py:226

↓ 1 callersFunctionmap_name

(name, opt)

src/data/conceptnet.py:10

↓ 1 callersFunctionmap_name

(name)

src/data/atomic.py:14

↓ 1 callersMethodmerge_heads

(self, x)

src/models/gpt.py:110

↓ 1 callersFunctionprint_atomic_sequence

(sequence_object)

src/interactive/functions.py:134

↓ 1 callersFunctionprint_category_help

(data)

src/interactive/functions.py:353

↓ 1 callersFunctionprint_conceptnet_sequence

(sequence_object)

src/interactive/functions.py:321

↓ 1 callersMethodprint_result

(self, split, epoch_losses)

src/evaluate/atomic_evaluate.py:36

↓ 1 callersFunctionreplace_params

(base_config, changes)

utils/utils.py:125

↓ 1 callersMethodreset_sequences

(self)

src/evaluate/conceptnet_generate.py:24

↓ 1 callersFunctionsave_checkpoint

(state, filename)

src/data/data.py:18

↓ 1 callersFunctionsave_sequences

(opt, sequences, avg_scores, indiv_scores, l, split, full, data_loader)

src/evaluate/generate.py:28

↓ 1 callersFunctionscore

(term1, term2, words, We, rel, Rel, Weight, Offset, evaType)

scripts/classify/demo_bilinear.py:36

↓ 1 callersFunctionscore

(hyp, refs)

scripts/evaluate/bleu_atomic.py:92

↓ 1 callersFunctionselect_partial_dataset

(data_opts, data)

src/data/atomic.py:57

↓ 1 callersFunctionset_atomic_inputs

(input_event, category, data_loader, text_encoder)

src/interactive/functions.py:149

↓ 1 callersFunctionset_conceptnet_inputs

(input_event, relation, text_encoder, max_e1, max_r, force)

src/interactive/functions.py:212

↓ 1 callersFunctionset_conceptnet_inputs_for_evaluation

(input_event, relation, output, text_encoder, max_e1, max_r, max_e2)

src/interactive/functions.py:235

↓ 1 callersMethodset_generator

(self, opt, model, data_loader)

src/train/conceptnet_train.py:22

↓ 1 callersFunctionset_progress_bar

(num_examples)

utils/utils.py:139

next →1–100 of 222, ranked by callers