MCPcopy Create free account

hub / github.com/NZWANG/MM-GTUNets / functions

Functions81 in github.com/NZWANG/MM-GTUNets

↓ 4 callersMethod__init__
(self, args)
model/mm_gtunets.py:12
↓ 4 callersMethodload_tensor
(self, load_path)
utils/mydataloader.py:192
↓ 4 callersMethodsave_tensor
(self, tensor, name, save_folder)
utils/mydataloader.py:187
↓ 3 callersFunctionaccuracy
Accuracy, auc with masking.Acc of the masked samples
utils/metrics.py:32
↓ 3 callersFunctionget_fc
(file_name, variable, norm)
utils/tools.py:150
↓ 3 callersMethodget_sub_scores
(self, subject_list, score, encode=True)
utils/mydataloader.py:88
↓ 3 callersMethodget_subject_IDs
(self, id_file="id.txt", num_subjects=None)
utils/mydataloader.py:68
↓ 3 callersMethodget_timeseries
(self, subject_list)
utils/mydataloader.py:72
↓ 3 callersMethodinitialize
(self)
opt.py:122
↓ 2 callersFunctionauc
input: logits, labels
utils/metrics.py:46
↓ 2 callersMethodcal_attention_score
(self, attention, shared_attention)
model/mm_gtunets.py:117
↓ 2 callersMethodcal_smh_loss
(self, embed, L)
model/mm_gtunets.py:252
↓ 2 callersMethodcalculate_connectivity
(self, timeseries, subject, kind='correlation', save=True)
utils/mydataloader.py:76
↓ 2 callersFunctionevaluate
()
train_mm_gtunets.py:83
↓ 2 callersMethodget_labels
(self, subject_list, binary=True)
utils/mydataloader.py:96
↓ 2 callersFunctionget_subject_score
Obtain phenotypic information of subjects.
utils/tools.py:217
↓ 2 callersFunctionmetrics
(preds, labels)
utils/metrics.py:67
↓ 2 callersMethodsave_checkpoint
Saves model when validation loss decrease.
utils/tools.py:64
↓ 1 callersMethodcal_deg_loss
(self)
model/mm_gtunets.py:257
↓ 1 callersFunctioncal_feature_sim
(features, tensor=True, self_loop=True)
utils/tools.py:237
↓ 1 callersMethodcal_graph_loss
(self, img_embed, ph_embed)
model/mm_gtunets.py:232
↓ 1 callersMethodcal_reward_loss
(self)
model/mm_gtunets.py:263
↓ 1 callersMethodcal_value
(self, graphs, reward, penalty)
model/mm_gtunets.py:101
↓ 1 callersFunctionchoose_atlas
(atlas)
opt.py:36
↓ 1 callersFunctionchoose_dataset
(dataset)
opt.py:8
↓ 1 callersFunctioncreate_id_file
Create the ID file.
utils/tools.py:88
↓ 1 callersFunctioncreate_reward_penalty_graph
(ph_dict, labels, train_idx, val_idx, test_idx, args)
model/rp_graph.py:4
↓ 1 callersMethodcreate_rp_graph
(self, affinity_graphs)
model/mm_gtunets.py:227
↓ 1 callersMethoddata_split
(self, n_folds, val_ratio=0)
utils/mydataloader.py:175
↓ 1 callersFunctionfeature_selection
matrix : feature matrix (num_subjects x num_features) labels : ground truth labels (num_subjects x 1) train_ind
utils/tools.py:292
↓ 1 callersMethodfetch_abide
(self, derivatives)
utils/mydataloader.py:126
↓ 1 callersMethodforward
(self, x)
model/mm_gtunets.py:39
↓ 1 callersFunctionget_ids
Obtain the ID of subjects.
utils/tools.py:72
↓ 1 callersFunctionget_networks
(subject_list, args, norm=True, kind='correlation')
utils/tools.py:162
↓ 1 callersMethodget_networks
(self, subject_list, norm=True, kind='correlation')
utils/mydataloader.py:80
↓ 1 callersMethodget_phenotypic_data
(self, subject_list, score_list)
utils/mydataloader.py:105
↓ 1 callersFunctionget_timeseries
(subject_list, args)
utils/tools.py:124
↓ 1 callersFunctionget_upper_triangle_networks
(subject_list, args, norm=True, kind='correlation')
utils/tools.py:190
↓ 1 callersMethodget_upper_triangle_networks
(self, subject_list, norm=True)
utils/mydataloader.py:84
↓ 1 callersMethodinit_vae_early_stop
(self)
model/mm_gtunets.py:182
↓ 1 callersMethodinit_vae_optimizer
(self)
model/mm_gtunets.py:178
↓ 1 callersMethodinit_vae_save_path
(self)
model/mm_gtunets.py:175
↓ 1 callersMethodload_data
(self, save=False)
utils/mydataloader.py:28
↓ 1 callersMethodload_vae
(self)
model/mm_gtunets.py:185
↓ 1 callersFunctionmove_files_to_main_directory
(main_directory)
utils/tools.py:318
↓ 1 callersFunctionnumeric_score
(pred, gt)
utils/metrics.py:59
↓ 1 callersFunctionordinal_encoding
Map string data to integer, filling missing values with -1.
utils/tools.py:264
↓ 1 callersMethodprint_args
(self)
opt.py:112
↓ 1 callersFunctionprint_result
(opt, n_folds, accs, sens, spes, aucs, f1)
utils/tools.py:352
↓ 1 callersMethodprocess_abide
(self, kind='correlation')
utils/mydataloader.py:134
↓ 1 callersMethodreparameterize
(self, mu, logvar)
model/mm_gtunets.py:34
↓ 1 callersFunctionreplace_non_floats
(file_name)
utils/tools.py:334
↓ 1 callersMethodreset
(self)
utils/metrics.py:19
↓ 1 callersMethodreset_parameters
r"""Resets all learnable parameters of the module.
model/gtunet.py:59
↓ 1 callersFunctionsave_result
(opt, n_folds, accs, sens, spes, aucs, f1)
utils/tools.py:361
↓ 1 callersMethodset_seed
(self, seed=0)
opt.py:126
↓ 1 callersMethodsetup_attenton
(self)
model/mm_gtunets.py:171
↓ 1 callersMethodsetup_vae_pretrain
(self)
model/mm_gtunets.py:165
↓ 1 callersFunctionsubject_connectivity
(timeseries, subject, args, kind='correlation', save=True)
utils/tools.py:107
↓ 1 callersFunctiontrain
()
train_mm_gtunets.py:13
↓ 1 callersMethodtrain_vae
(self, ph_features)
model/mm_gtunets.py:193
FunctionPSNR
(mse, peak=1.)
utils/metrics.py:10
Method__call__
(self, val_loss, model)
utils/tools.py:45
Method__init__
(self, model=None, dataset="ABIDE", atlas="aal")
opt.py:45
Method__init__
(self, args, features=None, labels=None, ph_dict=None, ph_data=None)
utils/mydataloader.py:13
Method__init__
Args: patience (int): How long to wait after last time validation loss improved. Default: 7
utils/tools.py:20
Method__init__
(self)
utils/metrics.py:16
Method__init__
( self, in_channels: int, hidden_channels: int, out_channels:
model/gtunet.py:13
Method__init__
(self, in_channels, out_channels)
model/mm_gtunets.py:54
Method__init__
(self, args)
model/mm_gtunets.py:68
Method__init__
(self, args)
model/mm_gtunets.py:110
Method__init__
(self, args, fold)
model/mm_gtunets.py:143
Method__repr__
(self)
model/gtunet.py:117
Methodforward
(self, x: Tensor, edge_index: Tensor, edge_weight: OptTensor = None, batch: Op
model/gtunet.py:68
Methodforward
(self, x)
model/mm_gtunets.py:59
Methodforward
(self, affinity_graphs)
model/mm_gtunets.py:84
Methodforward
(self, img_embed, ph_embed)
model/mm_gtunets.py:122
Methodforward
(self, img_features, ph_features, affinity_graphs)
model/mm_gtunets.py:267
Functionprf
input: logits, labels
utils/metrics.py:38
Methodprocess_adhd200
(self, kind='correlation', filter=True)
utils/mydataloader.py:145
Methodupdate
(self, val, n=1)
utils/metrics.py:25