Functions — github.com/debjitpaul/refiner

↓ 1 callersMethodpredict

(self, test_file, per_gpu_eval_batch_size, max_generated_tokens)

src/refiner.py:385

↓ 1 callersFunctionreload_answer_validity

()

docs/static/js/critic_model.js:13

↓ 1 callersMethodremove_operations

(self, equation)

data_preprocessing/create_negative_examples_mwp.py:83

↓ 1 callersFunctionsave_dataset

(dataset, path)

data_preprocessing/prepare_snlr_datasets.py:41

↓ 1 callersFunctionswap

(r, index, count)

src/eval/conala_eval.py:154

↓ 1 callersMethodtrain

Train the model

src/exp_t5_lm.py:101

↓ 1 callersMethodtrain

Train the model

src/refiner.py:98

↓ 1 callersFunctiontraining

(training_file, dev_file, trained_models_dir, trained_critique_dir, per

src/scripts/test_predict.py:12

↓ 1 callersFunctiontraining

(training_file, dev_file, trained_models_dir, per_gpu_train_batch_size,

src/scripts/finetune.py:19

↓ 1 callersFunctiontraining

(training_file, dev_file, trained_models_dir, trained_critique_dir, per

src/scripts/train_refiner.py:17

↓ 1 callersMethodtree2linear

- + number1 number2 number3 --> add(number1, number2) | subtract(#0, number3) - number1 + number2 number3 --> add(number2, number3) |

docs/services/data_service.py:86

Method__getitem__

(self, index)

data_preprocessing/utils.py:166

Method__init__

(self, max_seq_length, output_model_dir, output_critique_mo

src/exp_t5_lm.py:39

Method__init__

(self, max_seq_length, output_model_dir, pretrained_model_n

src/finetune.py:26

Method__init__

(self, max_seq_length, output_model_dir, output_critique_mo

src/test_lm.py:39

Method__init__

(self, max_seq_length, output_model_dir, output_critique_mo

src/refiner.py:30

Method__init__

(self, dataset, batch_size)

src/data_processing/multi_task_batch_scheduler.py:12

Method__init__

Initialisation of actor critic model Args: actor_model_dir (_type_): directory of actor pretrained model critic_model

docs/services/model_service.py:10

Method__init__

init from file

docs/services/data_service.py:8

Method__init__

(self, data, tokenizer, source_col="input", target_col="target", max_source_len=512, max_target_len=512)

data_preprocessing/utils.py:155

Method__init__

(self, data_file)

data_preprocessing/create_negative_examples_mwp.py:10

Method__iter__

(self)

src/data_processing/multi_task_batch_scheduler.py:21

Method__len__

(self)

src/data_processing/multi_task_batch_scheduler.py:18

Method__len__

(self)

data_preprocessing/utils.py:163

Method_add_gen_input

(self, gen, input_ids, label)

src/test_lm.py:726

Method_add_gen_input

(self, gen, input_ids, label)

src/refiner.py:669

Method_critique_function_batch