Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/aikunyi/FourierGNN
/ functions
Functions
28 in github.com/aikunyi/FourierGNN
⨍
Functions
28
◇
Types & classes
5
↓ 4 callers
Function
MAE
Mean absolute error. :param v: np.ndarray or int, ground truth. :param v_: np.ndarray or int, prediction. :param axis: axis to do
utils/utils.py:151
↓ 4 callers
Function
MAPE
Mean absolute percentage error. :param v: np.ndarray or int, ground truth. :param v_: np.ndarray or int, prediction. :param axis:
utils/utils.py:118
↓ 4 callers
Function
RMSE
Mean squared error. :param v: np.ndarray or int, ground truth. :param v_: np.ndarray or int, prediction. :param axis: axis to do
utils/utils.py:140
↓ 2 callers
Function
evaluate
:param y: array in shape of [count, time_step, node]. :param y_hat: in same shape with y. :param by_step: evaluate by time_step dim.
utils/utils.py:162
↓ 1 callers
Method
fourierGC
(self, x, B, N, L)
model/FourierGNN.py:49
↓ 1 callers
Function
load_model
(model_dir, epoch=None)
utils/utils.py:85
↓ 1 callers
Function
save_model
(model, model_dir, epoch=None)
utils/utils.py:74
↓ 1 callers
Method
tokenEmb
(self, x)
model/FourierGNN.py:43
↓ 1 callers
Function
validate
(model, vali_loader)
main.py:103
Method
__getitem__
(self, index)
data/data_loader.py:38
Method
__getitem__
(self, index)
data/data_loader.py:92
Method
__getitem__
(self, index)
data/data_loader.py:170
Method
__getitem__
(self, index)
data/data_loader.py:228
Method
__init__
(self, pre_length, embed_size, feature_size, seq_length, hidden_size, hard_thresholding_frac
model/FourierGNN.py:6
Method
__init__
(self, root_path, flag, seq_len, pre_len, type, train_ratio, val_ratio)
data/data_loader.py:10
Method
__init__
(self, root_path, flag, seq_len, pre_len, type, train_ratio, val_ratio)
data/data_loader.py:63
Method
__init__
(self, root_path, flag, seq_len, pre_len, type, train_ratio, val_ratio)
data/data_loader.py:118
Method
__init__
(self, root_path, flag, seq_len, pre_len, type, train_ratio, val_ratio)
data/data_loader.py:194
Method
__len__
(self)
data/data_loader.py:53
Method
__len__
(self)
data/data_loader.py:108
Method
__len__
(self)
data/data_loader.py:185
Method
__len__
(self)
data/data_loader.py:245
Function
concat_fun
(inputs, axis=-1)
utils/utils.py:13
Method
forward
(self, x)
model/FourierGNN.py:112
Function
masked_MAPE
Mean absolute percentage error. :param v: np.ndarray or int, ground truth. :param v_: np.ndarray or int, prediction. :param axis:
utils/utils.py:98
Function
slice_arrays
Slice an array or list of arrays. This takes an array-like, or a list of array-likes, and outputs: - arrays[start:stop] if `array
utils/utils.py:20
Function
smape
(P, A)
utils/utils.py:133
Function
test
()
main.py:128