Functions — github.com/ZhuYun97/ENGINE

CodeHub MCPcopy Create free account

hub / github.com/ZhuYun97/ENGINE / functions

Functions97 in github.com/ZhuYun97/ENGINE

⨍ Functions 97 ◇ Types & classes 17

↓ 21 callersFunctionget_norm

models/encoder.py:25

↓ 12 callersMethodreset_parameters

models/encoder.py:85

↓ 11 callersMethodeval

models/LMs/trainer.py:162

↓ 9 callersMethodtrain

models/LMs/trainer.py:81

↓ 7 callersFunctionget_activation

models/encoder.py:13

↓ 6 callersMethod__init__

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:36

↓ 6 callersFunctioncollect_subgraphs

(selected_id, graph, walk_steps=20, restart_ratio=0.5)

data/sampling.py:25

↓ 6 callersFunctionego_graphs_sampler

(node_idx, data, hop=2, sparse=False)

data/sampling.py:86

↓ 6 callersFunctionload_data

(dataset, use_dgl=False, use_text=False, use_gpt=False, seed=0)

↓ 5 callersFunctioneval

(test_loader, xs, model_list, prog_list, alpha_list)

↓ 5 callersMethodparse_args

utils/args.py:56

↓ 3 callersFunctioncollect_txt

↓ 3 callersFunctioncollect_txt

finetune_lm.py:12

↓ 3 callersFunctioneval_fullgraph

(model, data, device, config, eval="valid")

traditional_gnn.py:93

↓ 3 callersFunctioneval_subgraph

(model, data_loader, device)

traditional_gnn.py:47

↓ 2 callersMethod__init__

(self, model, n_labels, dropout=0.0, seed=0, cla_bias=True, feat_shrink='')

models/LMs/model.py:10

↓ 2 callersFunctionefficient_eval

(test_loader, xs, model_list, prog_list, alpha_list, exit_list)

↓ 2 callersMethodforward_subgraph

(self, x, edge_index, batch, root_n_id, edge_weight=None, **kwargs)

models/gnn.py:32

↓ 2 callersFunctionget_cur_time

(timezone='Asia/Shanghai', t_format='%m-%d %H:%M:%S')

utils/time.py:19

↓ 1 callersFunctionadjust_idx

re-index the nodes and edge index In the subgraphs, some nodes are droppped. We need to change the node index in edge_index in order to correspon

data/sampling.py:72

↓ 1 callersFunctioncreate_peft_config

(model, method='lora')

utils/peft.py:4

↓ 1 callersFunctionefficient_train_eval

(train_loader, val_loader, test_loader, xs, model_list, prog_list, alpha_list, exit_list, optimizer)

↓ 1 callersFunctionget_cora_casestudy

data/data_utils/load_cora.py:12

↓ 1 callersFunctionget_dataloader

↓ 1 callersFunctionget_hidden_states

↓ 1 callersFunctionget_raw_text_products

(use_text=False, seed=0)

data/data_utils/load_products.py:52

↓ 1 callersFunctionload_llm

↓ 1 callersFunctionload_model

(input_dim: int, output_dim: int, config)

models/model_manager.py:4

↓ 1 callersFunctionmain

traditional_gnn.py:114

↓ 1 callersFunctionmean_pooling

(model_output, attention_mask)

↓ 1 callersFunctionparse_cora

data/data_utils/load_cora.py:60

↓ 1 callersFunctionsave_hidden_states

(path, max_length=512, llm_model='llama')

↓ 1 callersFunctionsubsampling

(data, config, sampler='rw')

utils/sampling.py:7

↓ 1 callersFunctiontime2str

utils/time.py:8

↓ 1 callersFunctiontrain_eval

(model, optimizer, criterion, config, data, train_loader, val_loader, test_loader, device)

traditional_gnn.py:107

↓ 1 callersFunctiontrain_eval

(train_loader, val_loader, test_loader, xs, model_list, prog_list, alpha_list, exit_list, optimizer)

↓ 1 callersFunctiontrain_fullgraph

(model, optimizer, criterion, config, data, device)

traditional_gnn.py:60

↓ 1 callersFunctiontrain_subgraph

(model, optimizer, criterion, config, train_loader, val_loader, test_loader, device)

traditional_gnn.py:15

Method__getitem__

data/dataset.py:9

(self, profiler)

(self, profiler)

finetune_lm.py:50

utils/register.py:9

utils/args.py:4

(self, encodings, labels)

data/dataset.py:5

(self, input_dim, layer_num=2, hidden_size=128, output_dim=70, activation="relu", dropout=0.5, norm='id', **ka

models/gnn.py:9

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:93

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:150

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:206

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:268

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:318

(self, input_dim, layer_num=2, hidden_size=128, output_dim=128, activation="relu", dropout=0.5, norm='id', las

models/encoder.py:372

models/LMs/trainer.py:20

(self, model, n_labels, dropout=0.0, seed=0, cla_bias=True, feat_shrink='')

models/LMs/model.py:55

(self, model, emb, pred, feat_shrink='')

models/LMs/model.py:98

data/dataset.py:14

Function_process

data/data_utils/load_products.py:16

Functionadd_remaining_selfloop_for_isolated_nodes

(edge_index, num_nodes)

data/sampling.py:10

Functioncollect_txt

Functioncompute_metrics

models/LMs/trainer.py:11

Methoddataloader_register

r""" Register for dataloader access. Args: dataloader_class (class): dataloader class Returns (class):

utils/register.py:88

Methoddataset_register

r""" Register for dataset access. Args: dataset_class (class): dataset class Returns (class): datase

utils/register.py:74

Methodencoder_register

r""" Register for model access. Args: model_class (class): model class Returns (class): model class

utils/register.py:60

Methodeval_and_save

models/LMs/trainer.py:125

Methodevaluator

(preds, labels)

models/LMs/trainer.py:157

(self, x, edge_index, edge_weight=None, frozen=False, **kwargs)

models/gnn.py:21

(self, x, edge_index, edge_weight=None)

models/encoder.py:66

(self, x, edge_index, edge_weight=None)

models/encoder.py:123

(self, x, edge_index, edge_weight=None)

models/encoder.py:179

(self, x, edge_index, **kwargs)

models/encoder.py:245

(self, x, edge_index, **kwargs)

models/encoder.py:299

(self, x, edge_index=None, **kwargs)

models/encoder.py:349

(self, x, edge_index, **kwargs)

models/encoder.py:403

(self, input_ids=None, attention_mask=None, neighbor_mask=None

models/LMs/model.py:25

(self, input_ids=None, attention_mask=None, labels=None,

models/LMs/model.py:71

(self, input_ids=None, attention_mask=None, labels=None,

models/LMs/model.py:107

Functionget_raw_text_arxiv

(use_text=False, seed=0)

data/data_utils/load_arxiv.py:8

Functionget_raw_text_arxiv_2023

(use_text=False, seed=0)

data/data_utils/load_arxiv_2023.py:8

Functionget_raw_text_citeseer

(use_text=False, seed=0)

data/data_utils/load_citeseer.py:10

Functionget_raw_text_cora

(use_text=False, seed=0)

data/data_utils/load_cora.py:81

Functionget_raw_text_photo

(use_text=False, seed=0)

data/data_utils/load_photo.py:8

Functionget_raw_text_wikics

(use_text=False, seed=0)

data/data_utils/load_wikics.py:5

Methodlauncher_register

r""" Register for pipeline access. Args: launcher_class (class): pipeline class Returns (class): pip

utils/register.py:32

Functionmean_pooling

(model_output, attention_mask)

finetune_lm.py:19

Methodmodel_register

r""" Register for model access. Args: model_class (class): model class Returns (class): model class

utils/register.py:46

Methodon_step_end

(self, *args, **kwargs)

Methodon_step_end

(self, *args, **kwargs)

finetune_lm.py:53

Methodpipeline_register

r""" Register for pipeline access. Args: pipeline_class (class): pipeline class Returns (class): pip

utils/register.py:18

Functionprocess_text

Methodreset_classifier

models/gnn.py:42

Methodreset_parameters

models/encoder.py:142

Methodreset_parameters

models/encoder.py:198

Methodreset_parameters

models/encoder.py:260

Methodreset_parameters

models/encoder.py:310

Methodreset_parameters

models/encoder.py:364

Methodreset_parameters

models/encoder.py:417

Functiontime_logger

utils/time.py:23

Functionwrapper

utils/time.py:24