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

↓ 1 callersMethodshuffle_sequences

(self, split="train", keys=None)

src/data/conceptnet.py:250

↓ 1 callersMethodshuffle_sequences

(self, split="train", keys=None)

src/data/atomic.py:186

↓ 1 callersFunctionsigmoid

(x)

scripts/classify/demo_bilinear.py:32

↓ 1 callersMethodstep

Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model

src/train/opt.py:57

↓ 1 callersFunctionupdate_indiv_generation_losses

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

src/evaluate/utils.py:24

↓ 1 callersFunctionupdate_indiv_generation_losses

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

src/train/utils.py:28

↓ 1 callersFunctionupdate_tensor_generation_losses

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

src/evaluate/utils.py:34

↓ 1 callersFunctionupdate_tensor_generation_losses

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

src/train/utils.py:40

↓ 1 callersMethodvalidate

(self, l, split="dev", losses={}, keyset=None)

src/evaluate/evaluate.py:22

(self, opt, model, data_loader)

src/evaluate/atomic_evaluate.py:13

Method__init__

(self, opt, model, data_loader)

src/evaluate/conceptnet_generate.py:15

Method__init__

(self, opt, model, data_loader, scorers, reward_function=None)

src/evaluate/generate.py:56

Method__init__

(self, opt, model, data_loader, track=False)

src/evaluate/conceptnet_evaluate.py:14

Method__init__

(self, opt, data_loader, batch_mode=True)

src/evaluate/sampler.py:27

Method__init__

(self, opt, data_loader, batch_mode=True)

src/evaluate/sampler.py:98

Method__init__

(self, opt, data_loader, batch_mode=True, scorer=None)

src/evaluate/sampler.py:182

Method__init__

(self, opt, model, data_loader)

src/evaluate/evaluate.py:9

Method__init__

(self, encoder_path, bpe_path)

src/data/utils.py:59

Method__init__

(self, opt, categories=None)

src/data/conceptnet.py:71

Method__init__

(self, opt, categories)

src/data/atomic.py:63

Method__init__

(self, n_state, e=1e-5)

(self, nx, n_ctx, cfg, scale=False)

src/models/gpt.py:77

Method__init__

(self, n_ctx, cfg, scale=False)

src/models/gpt.py:152

Method__init__

(self, cfg, vocab=40990, n_ctx=512)

src/models/gpt.py:171

Method__init__

(self, cfg, vocab=40990, n_ctx=512, return_probs=False, return_acts=False)

src/models/gpt.py:194

Method__init__

(self, model, cfg, trunc_and_reshape=True)

src/models/gpt.py:219

Method__init__

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

src/train/train.py:178

Method__init__

(self, params, lr, schedule, warmup, t_total, b1=0.9, b2=0.999, e=1e-8, l2=0,

src/train/opt.py:37

Method__init__

(self, opt, *args)

src/train/atomic_train.py:14

src/train/conceptnet_train.py:26

Functionbatch_atomic_generate

(opt, nums, losses, batch_variables, eval_mode=False)

src/train/batch.py:21

Functionbatch_conceptnet_generate

(opt, nums, losses, batch_variables, eval_mode=False, tracking_mode=False)

src/train/batch.py:58

Methodcompute_final_scores

(self, average_loss, nums)

src/evaluate/conceptnet_evaluate.py:52

Methodcounter

(self, nums)

src/evaluate/conceptnet_evaluate.py:73

Methoddecide_to_save

(self)

src/train/conceptnet_train.py:57

Functiondo_gen_run

(opt, generator, l, split="dev", scores={})

src/evaluate/generate.py:6

Functionevaluate_conceptnet_sequence

(e1, model, data_loader, text_encoder, relation, e2, force=False)

src/interactive/functions.py:262

scripts/evaluate/bleu_atomic.py:44

(self, x, sequence_mask)

src/models/gpt.py:123

Methodforward

(self, x)

src/models/gpt.py:145

Methodforward

(self, x, sequence_mask)

src/models/gpt.py:160

Methodforward

(self, x, sequence_mask)

src/models/gpt.py:182

Methodforward

(self, x, sequence_mask=None)

src/models/gpt.py:206

Methodforward

(self, h)

src/models/gpt.py:227

src/evaluate/conceptnet_generate.py:27

Methodgenerate_batch

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

src/evaluate/generate.py:70

Functiongenerate_config_files

(type_, key, name="base", eval_mode=False)

utils/utils.py:95

Methodgenerate_sequence

(self, batch, model, data_loader, start_idx, end_len)

src/evaluate/sampler.py:36

Methodgenerate_sequence

(self, batch, model, data_loader, start_idx, end_len)

src/evaluate/sampler.py:107

Methodgenerate_sequence

(self, batch, model, data_loader, start_idx, end_len)

src/evaluate/sampler.py:205

Functionget_atomic_sequence

(input_event, model, sampler, data_loader, text_encoder, category)

src/interactive/functions.py:93

Functionget_conceptnet_sequence

(e1, model, sampler, data_loader, text_encoder, relation, force=False)

src/interactive/functions.py:163

Functionget_parameters

(opt, exp_type="model")

src/data/config.py:14

Methodinitialize_losses

(self)

src/evaluate/conceptnet_evaluate.py:39

Functioninitialize_progress_bar

(data_loader_list)

utils/utils.py:133

Functionload_checkpoint

src/data/config.py:183

Functionload_data

(dataset, opt)

src/interactive/functions.py:21

Methodload_data

(self, path)

src/data/conceptnet.py:89

Functionload_existing_data_loader

(data_loader, path)

src/data/utils.py:10

Functionload_model_file

(model_file)

src/interactive/functions.py:14

Functionload_state_dict

(model, state_dict)

src/models/models.py:27

Functionmain

(num)

src/main_atomic.py:17

Functionmake_batch

(X)

scripts/generate/generate_atomic_topk.py:117

Functionmake_batch

(X)

scripts/generate/generate_atomic_beam_search.py:128

Functionmake_batch

(X)

scripts/generate/generate_conceptnet_beam_search.py:143

Functionmake_batch

(X)

scripts/generate/generate_atomic_greedy.py:127

Methodmake_batch

(self, X)

src/evaluate/sampler.py:188

Functionmake_data_loader

(opt, *args)

src/data/data.py:70

Functionmake_evaluator

(opt, *args)

src/evaluate/atomic_evaluate.py:5

Functionmake_evaluator

(opt, *args, **kwargs)

src/evaluate/conceptnet_evaluate.py:9

Functionmake_generator

(opt, *args)

src/evaluate/conceptnet_generate.py:10

Functionmake_model

(opt, n_vocab, n_ctx, state_dict)

src/interactive/functions.py:67

Functionmake_model

(opt, n_vocab, n_ctx, n_special, load=True, return_acts=True, return_probs=False,

src/models/models.py:5

Functionmake_name

(opt, prefix="", eval_=False, is_dir=True, set_epoch=None, do_epoch=True)

utils/utils.py:24

Functionmake_new_tensor_from_list

(items, device_num, dtype=torch.float32)

utils/utils.py:14

Functionmake_sampler

(sampler_type, opt, *args, **kwargs)

src/evaluate/sampler.py:11

Methodmake_tensors

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

src/data/conceptnet.py:143

Functionmake_trainer

(opt, *args)

src/train/conceptnet_train.py:12

Functionmake_trainer

(opt, *args)

src/train/atomic_train.py:9

Functionmerge_list_of_dicts

(L)

utils/utils.py:145

Functionmodify_output_for_loss_fn

(loss_fn, output, dim)

src/train/utils.py:50

Functionmulti_gpu

(model, devices)

src/models/models.py:23

Methodoffset_summary

(self, split)

src/data/conceptnet.py:86

Functionprepare_position_embeddings

(opt, encoder_vocab, sequences)

src/models/utils.py:4

Functionprint_help

(data)

src/interactive/functions.py:336

Functionprint_relation_help

(data)

src/interactive/functions.py:349

Methodprint_result

(self, split, epoch_losses)

src/evaluate/conceptnet_evaluate.py:76

Functionprint_sampling_help

()

src/interactive/functions.py:404

Functionprune_data_for_evaluation

(data_loader, categories, split)

src/data/atomic.py:207

← previous next →101–200 of 222, ranked by callers