MCPcopy Create free account

hub / github.com/ViktorAxelsen/TFE-GNN / functions

Functions29 in github.com/ViktorAxelsen/TFE-GNN

↓ 4 callersFunctionconstruct_graph_format_data
(file_path, save_path, type, w_size=generalConfig.PMI_WINDOW_SIZE, pmi=1)
preprocess.py:60
↓ 4 callersFunctionshow_time
()
utils.py:19
↓ 3 callersMethodstep
(self, epoch=None)
optim.py:10
↓ 2 callersMethod__init__
(self, embedding_size, h_feats, dropout)
model.py:12
↓ 2 callersFunctionconstruct_dataset_from_bytes_ISCX
(dir_path_dict, type)
preprocess.py:14
↓ 2 callersFunctionget_bytes_from_raw
(s)
utils.py:47
↓ 2 callersFunctionget_device
(index=3)
utils.py:15
↓ 2 callersFunctionpad_truncate
(flow, type, config)
utils.py:61
↓ 2 callersFunctionremove
(flow)
utils.py:84
↓ 2 callersFunctionset_seed
(seed=config.SEED)
utils.py:25
↓ 1 callersFunctionconstruct_graph
(bytes, w_size, k=1)
utils.py:206
↓ 1 callersFunctionpcap2npy4ISCX
(dir_path_dict, save_path_dict)
pcap2npy.py:52
↓ 1 callersFunctionsplit_flow_ISCX
(file_path, cate, allow_empty, pad_trunc, config, type='payload')
utils.py:99
↓ 1 callersFunctionsplit_flow_Tor_nonoverlapping
(file_path, cate, allow_empty, pad_trunc, config, type='payload')
utils.py:141
↓ 1 callersFunctiontest
()
test.py:16
↓ 1 callersFunctiontrain
()
train.py:16
Method__getitem__
(self, index)
dataloader.py:23
Method__init__
(self, optimizer, warmup_iter, after_scheduler)
optim.py:5
Method__init__
(self, payload_path, header_path)
dataloader.py:11
Method__init__
(self, in_size)
model.py:35
Method__init__
(self, num_classes, embedding_size=64, h_feats=128, dropout=0.2, downstream_dropout=0.0)
model.py:58
Method__len__
(self)
dataloader.py:28
Methodforward
(self, g, in_feat)
model.py:21
Methodforward
(self, x, y)
model.py:48
Methodforward
(self, header_graph_data, payload_graph_data, labels)
model.py:71
Methodget_lr
(self)
optim.py:19
Functionmix_collate_fn
(batch)
utils.py:36
Functionprocess
(pkt)
pcap2npy.py:24
Methodprocess
(self)
dataloader.py:16