Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ViktorAxelsen/TFE-GNN
/ functions
Functions
29 in github.com/ViktorAxelsen/TFE-GNN
⨍
Functions
29
◇
Types & classes
10
↓ 4 callers
Function
construct_graph_format_data
(file_path, save_path, type, w_size=generalConfig.PMI_WINDOW_SIZE, pmi=1)
preprocess.py:60
↓ 4 callers
Function
show_time
()
utils.py:19
↓ 3 callers
Method
step
(self, epoch=None)
optim.py:10
↓ 2 callers
Method
__init__
(self, embedding_size, h_feats, dropout)
model.py:12
↓ 2 callers
Function
construct_dataset_from_bytes_ISCX
(dir_path_dict, type)
preprocess.py:14
↓ 2 callers
Function
get_bytes_from_raw
(s)
utils.py:47
↓ 2 callers
Function
get_device
(index=3)
utils.py:15
↓ 2 callers
Function
pad_truncate
(flow, type, config)
utils.py:61
↓ 2 callers
Function
remove
(flow)
utils.py:84
↓ 2 callers
Function
set_seed
(seed=config.SEED)
utils.py:25
↓ 1 callers
Function
construct_graph
(bytes, w_size, k=1)
utils.py:206
↓ 1 callers
Function
pcap2npy4ISCX
(dir_path_dict, save_path_dict)
pcap2npy.py:52
↓ 1 callers
Function
split_flow_ISCX
(file_path, cate, allow_empty, pad_trunc, config, type='payload')
utils.py:99
↓ 1 callers
Function
split_flow_Tor_nonoverlapping
(file_path, cate, allow_empty, pad_trunc, config, type='payload')
utils.py:141
↓ 1 callers
Function
test
()
test.py:16
↓ 1 callers
Function
train
()
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
Method
forward
(self, g, in_feat)
model.py:21
Method
forward
(self, x, y)
model.py:48
Method
forward
(self, header_graph_data, payload_graph_data, labels)
model.py:71
Method
get_lr
(self)
optim.py:19
Function
mix_collate_fn
(batch)
utils.py:36
Function
process
(pkt)
pcap2npy.py:24
Method
process
(self)
dataloader.py:16