MCPcopy Create free account

hub / github.com/JIA-Lab-research/MR-GSM8K / functions

Functions27 in github.com/JIA-Lab-research/MR-GSM8K

⨍ Functions 27◇ Types & classes 5
↓ 2 callersFunctionmulti_thread_response_generation
(data_list, client, model_name, stop_token_ids, max_workers=5, type='eval')
scripts/eval_mr_gsm8k.py:152
↓ 2 callersFunctionrequest_openai
(client, prompt, model, max_retry=5, temperature=0., max_token=1024, stop_token_ids=None)
scripts/eval_mr_gsm8k.py:6
↓ 1 callersFunctionMCC_score
(tp, tn, fp, fn)
scripts/eval_mr_gsm8k.py:38
↓ 1 callersFunction_make_r_io_base
(f, mode: str)
scripts/train_math.py:38
↓ 1 callersFunction_tokenize_fn
Tokenize a list of strings.
scripts/train_math.py:121
↓ 1 callersFunctionconstruct_eval_stats
(basic_stats, correct_sol_num=1427, incorrect_sol_num=1573)
scripts/eval_mr_gsm8k.py:47
↓ 1 callersFunctionconstruct_k_shot_demos
(demo_path, num_shots)
scripts/eval_mr_gsm8k.py:103
↓ 1 callersFunctionget_eval_outputs
(all_problems, k_shot_demos, output_file_path, client, model_name, stop_token_ids, max_workers=5)
scripts/eval_mr_gsm8k.py:222
↓ 1 callersFunctionget_mr_score
(model_stat, w1=0.2, w2=0.3, w3=0.5)
scripts/eval_mr_gsm8k.py:41
↓ 1 callersFunctiongrade_eval_outputs
(eval_responses, output_file_path, client, model_name, max_workers)
scripts/eval_mr_gsm8k.py:171
↓ 1 callersFunctionjload
Load a .json file into a dictionary.
scripts/train_math.py:43
↓ 1 callersFunctionmain
()
scripts/eval_mr_gsm8k.py:281
↓ 1 callersFunctionmake_supervised_data_module
Make dataset and collator for supervised fine-tuning.
scripts/train_math.py:250
↓ 1 callersFunctionpreprocess
Preprocess the data by tokenizing.
scripts/train_math.py:145
↓ 1 callersFunctionprocess_eval_results
(grade_responses)
scripts/eval_mr_gsm8k.py:69
↓ 1 callersFunctionsafe_save_model_for_hf_trainer
Collects the state dict and dump to disk.
scripts/train_math.py:89
↓ 1 callersFunctionsingle_thread_eval_generation
(data, client, model_name, stop_token_ids)
scripts/eval_mr_gsm8k.py:114
↓ 1 callersFunctionsingle_thread_grade_generation
(data, client, model_name)
scripts/eval_mr_gsm8k.py:133
↓ 1 callersFunctionsmart_tokenizer_and_embedding_resize
Resize tokenizer and embedding. Note: This is the unoptimized version that may make your embedding size not be divisible by 64.
scripts/train_math.py:98
↓ 1 callersFunctiontrain
()
scripts/train_math.py:257
Method__call__
(self, instances: Sequence[Dict])
scripts/train_math.py:228
Method__getitem__
(self, i)
scripts/train_math.py:207
Method__init__
(self, data_args, tokenizer: transformers.PreTrainedTokenizer)
scripts/train_math.py:162
Method__len__
(self)
scripts/train_math.py:201
Methodget_input
(query)
scripts/train_math.py:186
Methodnaive__call__
(self, instances: Sequence[Dict])
scripts/train_math.py:216
Methodnaive__getitem__
(self, i)
scripts/train_math.py:204