Functions — github.com/RowitZou/DC-Match

CodeHub MCPcopy Create free account

hub / github.com/RowitZou/DC-Match / functions

Functions19 in github.com/RowitZou/DC-Match

⨍ Functions 19 ◇ Types & classes 4

↓ 4 callersMethod_pad

(self, data, width=-1, dtype=torch.long)

src/others/data_collator.py:13

↓ 2 callersMethodhandle_special_token

(self, token_list)

src/prepro/json_to_data.py:15

↓ 1 callersMethodapply

src/models/sm_model.py:52

↓ 1 callersFunctioninit_logger

(log_file=None, log_file_level=logging.NOTSET)

src/others/logging.py:9

↓ 1 callersFunctiontest

↓ 1 callersFunctiontrain

src/train.py:16

features: input_ids, token_type_ids, attention_mask, labels, keyword_mask, context_mask, special_mask, or

src/others/data_collator.py:19

(self, args, tokenizer, padding=True)

src/others/data_collator.py:8

src/others/metrics.py:7

src/prepro/json_to_data.py:10

(self, model_name, labels, loss_type=1, checkpoint=None, debug=False)

src/models/sm_model.py:9

Method_init_weights

Initialize the weights

src/models/sm_model.py:36

Methodcompute_metrics_f1

src/others/metrics.py:10

Methodcompute_metrics_macro_f1

src/others/metrics.py:18

Functionformat_json_to_data

src/prepro/json_to_data.py:88

(self, input_ids, token_type_ids, attention_mask, labels, keyword_mask, context_mask, special_

src/models/sm_model.py:58

(self, example)

src/prepro/json_to_data.py:18

Functionstr2bool

src/preprocess.py:8

Functionstr2bool