Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NZWANG/MM-GTUNets
/ functions
Functions
81 in github.com/NZWANG/MM-GTUNets
⨍
Functions
81
◇
Types & classes
10
↓ 4 callers
Method
__init__
(self, args)
model/mm_gtunets.py:12
↓ 4 callers
Method
load_tensor
(self, load_path)
utils/mydataloader.py:192
↓ 4 callers
Method
save_tensor
(self, tensor, name, save_folder)
utils/mydataloader.py:187
↓ 3 callers
Function
accuracy
Accuracy, auc with masking.Acc of the masked samples
utils/metrics.py:32
↓ 3 callers
Function
get_fc
(file_name, variable, norm)
utils/tools.py:150
↓ 3 callers
Method
get_sub_scores
(self, subject_list, score, encode=True)
utils/mydataloader.py:88
↓ 3 callers
Method
get_subject_IDs
(self, id_file="id.txt", num_subjects=None)
utils/mydataloader.py:68
↓ 3 callers
Method
get_timeseries
(self, subject_list)
utils/mydataloader.py:72
↓ 3 callers
Method
initialize
(self)
opt.py:122
↓ 2 callers
Function
auc
input: logits, labels
utils/metrics.py:46
↓ 2 callers
Method
cal_attention_score
(self, attention, shared_attention)
model/mm_gtunets.py:117
↓ 2 callers
Method
cal_smh_loss
(self, embed, L)
model/mm_gtunets.py:252
↓ 2 callers
Method
calculate_connectivity
(self, timeseries, subject, kind='correlation', save=True)
utils/mydataloader.py:76
↓ 2 callers
Function
evaluate
()
train_mm_gtunets.py:83
↓ 2 callers
Method
get_labels
(self, subject_list, binary=True)
utils/mydataloader.py:96
↓ 2 callers
Function
get_subject_score
Obtain phenotypic information of subjects.
utils/tools.py:217
↓ 2 callers
Function
metrics
(preds, labels)
utils/metrics.py:67
↓ 2 callers
Method
save_checkpoint
Saves model when validation loss decrease.
utils/tools.py:64
↓ 1 callers
Method
cal_deg_loss
(self)
model/mm_gtunets.py:257
↓ 1 callers
Function
cal_feature_sim
(features, tensor=True, self_loop=True)
utils/tools.py:237
↓ 1 callers
Method
cal_graph_loss
(self, img_embed, ph_embed)
model/mm_gtunets.py:232
↓ 1 callers
Method
cal_reward_loss
(self)
model/mm_gtunets.py:263
↓ 1 callers
Method
cal_value
(self, graphs, reward, penalty)
model/mm_gtunets.py:101
↓ 1 callers
Function
choose_atlas
(atlas)
opt.py:36
↓ 1 callers
Function
choose_dataset
(dataset)
opt.py:8
↓ 1 callers
Function
create_id_file
Create the ID file.
utils/tools.py:88
↓ 1 callers
Function
create_reward_penalty_graph
(ph_dict, labels, train_idx, val_idx, test_idx, args)
model/rp_graph.py:4
↓ 1 callers
Method
create_rp_graph
(self, affinity_graphs)
model/mm_gtunets.py:227
↓ 1 callers
Method
data_split
(self, n_folds, val_ratio=0)
utils/mydataloader.py:175
↓ 1 callers
Function
feature_selection
matrix : feature matrix (num_subjects x num_features) labels : ground truth labels (num_subjects x 1) train_ind
utils/tools.py:292
↓ 1 callers
Method
fetch_abide
(self, derivatives)
utils/mydataloader.py:126
↓ 1 callers
Method
forward
(self, x)
model/mm_gtunets.py:39
↓ 1 callers
Function
get_ids
Obtain the ID of subjects.
utils/tools.py:72
↓ 1 callers
Function
get_networks
(subject_list, args, norm=True, kind='correlation')
utils/tools.py:162
↓ 1 callers
Method
get_networks
(self, subject_list, norm=True, kind='correlation')
utils/mydataloader.py:80
↓ 1 callers
Method
get_phenotypic_data
(self, subject_list, score_list)
utils/mydataloader.py:105
↓ 1 callers
Function
get_timeseries
(subject_list, args)
utils/tools.py:124
↓ 1 callers
Function
get_upper_triangle_networks
(subject_list, args, norm=True, kind='correlation')
utils/tools.py:190
↓ 1 callers
Method
get_upper_triangle_networks
(self, subject_list, norm=True)
utils/mydataloader.py:84
↓ 1 callers
Method
init_vae_early_stop
(self)
model/mm_gtunets.py:182
↓ 1 callers
Method
init_vae_optimizer
(self)
model/mm_gtunets.py:178
↓ 1 callers
Method
init_vae_save_path
(self)
model/mm_gtunets.py:175
↓ 1 callers
Method
load_data
(self, save=False)
utils/mydataloader.py:28
↓ 1 callers
Method
load_vae
(self)
model/mm_gtunets.py:185
↓ 1 callers
Function
move_files_to_main_directory
(main_directory)
utils/tools.py:318
↓ 1 callers
Function
numeric_score
(pred, gt)
utils/metrics.py:59
↓ 1 callers
Function
ordinal_encoding
Map string data to integer, filling missing values with -1.
utils/tools.py:264
↓ 1 callers
Method
print_args
(self)
opt.py:112
↓ 1 callers
Function
print_result
(opt, n_folds, accs, sens, spes, aucs, f1)
utils/tools.py:352
↓ 1 callers
Method
process_abide
(self, kind='correlation')
utils/mydataloader.py:134
↓ 1 callers
Method
reparameterize
(self, mu, logvar)
model/mm_gtunets.py:34
↓ 1 callers
Function
replace_non_floats
(file_name)
utils/tools.py:334
↓ 1 callers
Method
reset
(self)
utils/metrics.py:19
↓ 1 callers
Method
reset_parameters
r"""Resets all learnable parameters of the module.
model/gtunet.py:59
↓ 1 callers
Function
save_result
(opt, n_folds, accs, sens, spes, aucs, f1)
utils/tools.py:361
↓ 1 callers
Method
set_seed
(self, seed=0)
opt.py:126
↓ 1 callers
Method
setup_attenton
(self)
model/mm_gtunets.py:171
↓ 1 callers
Method
setup_vae_pretrain
(self)
model/mm_gtunets.py:165
↓ 1 callers
Function
subject_connectivity
(timeseries, subject, args, kind='correlation', save=True)
utils/tools.py:107
↓ 1 callers
Function
train
()
train_mm_gtunets.py:13
↓ 1 callers
Method
train_vae
(self, ph_features)
model/mm_gtunets.py:193
Function
PSNR
(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
Method
forward
(self, x: Tensor, edge_index: Tensor, edge_weight: OptTensor = None, batch: Op
model/gtunet.py:68
Method
forward
(self, x)
model/mm_gtunets.py:59
Method
forward
(self, affinity_graphs)
model/mm_gtunets.py:84
Method
forward
(self, img_embed, ph_embed)
model/mm_gtunets.py:122
Method
forward
(self, img_features, ph_features, affinity_graphs)
model/mm_gtunets.py:267
Function
prf
input: logits, labels
utils/metrics.py:38
Method
process_adhd200
(self, kind='correlation', filter=True)
utils/mydataloader.py:145
Method
update
(self, val, n=1)
utils/metrics.py:25