MCPcopy Create free account

hub / github.com/SamHaoYuan/DSANForAAAI2021 / functions

Functions162 in github.com/SamHaoYuan/DSANForAAAI2021

⨍ Functions 162◇ Types & classes 25
Method__init__
(self, train_file, test_file, padding_idx=0)
baselines/pop.py:49
Method__init__
(self, session_index, session_id, items_indexes)
baselines/Bert4Rec.py:37
Method__init__
(self, train_file, test_file, padding_idx=0)
baselines/Bert4Rec.py:67
Method__init__
(self, hidden_size, activate="relu", head_num=2, dropout=0, initializer_range=0.02)
baselines/Bert4Rec.py:412
Method__init__
(self, hidden_size, initializer_range=0.02)
baselines/Bert4Rec.py:470
Method__init__
(self, hidden_size, activate="relu", head_num=2, dropout=0, attention_dropout=0, initializer_
baselines/Bert4Rec.py:484
Method__init__
(self, hidden_size=64, itemNum=0, posNum=0, padding_idx=0, dropout=0.5, attention_dropout=0, head_num=2,
baselines/Bert4Rec.py:521
Method__init__
(self, session_index, session_id, items_indexes)
baselines/STAMP.py:18
Method__init__
(self, train_file, test_file, padding_idx=0)
baselines/STAMP.py:48
Method__init__
(self, method="specific", hidden_size=64)
baselines/STAMP.py:353
Method__init__
(self, data, shuffle=False, graph=None)
baselines/GCSAN.py:35
Method__init__
(self, hidden_size, dropout=0)
baselines/GCSAN.py:174
Method__init__
(self, opt, n_node)
baselines/GCSAN.py:226
Method__init__
(self)
baselines/GCSAN.py:342
Method__init__
(self, session_id, session, session_timestamp, sample_size=0, k=500)
baselines/KNN-based/SKNN.py:6
Method__init__
(self, session_id, session, session_timestamp, sample_size=0, k=500, factor1=True, l1=3.54, f
baselines/KNN-based/STAN.py:6
Method__init__
(self, sessions, neighbors, neighbor_weights, labels, session_length=20, k=20, padding_idx=0)
baselines/CoSAN/dataloader.py:79
Method__init__
(self, hidden_dim, head, dropout=0)
baselines/CoSAN/CoSAN.py:8
Method__init__
(self, hidden_dim, dropout=0)
baselines/CoSAN/CoSAN.py:43
Method__init__
(self, hidden_dim, head, dropout=0)
baselines/CoSAN/CoSAN.py:55
Method__len__
(self)
baselines/pop.py:289
Method__len__
(self)
baselines/Bert4Rec.py:336
Method__len__
(self)
baselines/STAMP.py:247
Method__len__
(self)
baselines/CoSAN/dataloader.py:96
Method__str__
(self)
baselines/pop.py:42
Method__str__
(self)
baselines/Bert4Rec.py:60
Method__str__
(self)
baselines/STAMP.py:41
Functionbpr_loss
(r)
baselines/pop.py:293
Functionbpr_loss
(r)
baselines/Bert4Rec.py:340
Functionbpr_loss
(r)
baselines/STAMP.py:251
Methodforward
(self, encoder_outputs, mask=None)
baselines/Bert4Rec.py:444
Methodforward
(self, x)
baselines/Bert4Rec.py:477
Methodforward
(self, encoder_outputs, mask=None)
baselines/Bert4Rec.py:494
Methodforward
(self, compute_output, attention_mask)
baselines/Bert4Rec.py:514
Methodforward
(self, hidden, encoder_outputs=None, x_t=None, mask=None)
baselines/STAMP.py:426
Methodforward
(self, session)
baselines/STAMP.py:482
Methodforward
(self, A, hidden)
baselines/GCSAN.py:166
Methodforward
(self, encoder_outputs, mask=None, is_train=True)
baselines/GCSAN.py:207
Methodforward
(self, inputs, A)
baselines/GCSAN.py:261
Methodforward
(self, q, k, v, mask=None)
baselines/CoSAN/CoSAN.py:27
Methodforward
(self, x)
baselines/CoSAN/CoSAN.py:50
Methodforward
(self, x, mask)
baselines/CoSAN/CoSAN.py:64
Methodforward
:param is_train: train or test :param session_id: (b, ) :param session: (b, len) :param neighbor_id: (b, len, k)
baselines/CoSAN/CoSAN.py:89
Methodget_batch_tasks_with_neg
(self, batch_size, session_length=10, predict_length=1, all_data=None, phase="train",
baselines/Bert4Rec.py:240
Methodget_batch_with_neg
(self, batch_size, session_length=10, predict_length=1, all_data=None, phase="train",
baselines/Bert4Rec.py:207
Functionget_mrr
:param pre: (B,K) TOP-K indics predicted by the model :param truth: (B, 1) real label :return: MRR(Float), the mrr score
metric.py:25
Functionget_recall
:param pre: (B,K) TOP-K indics predicted by the model :param truth: (B,1) the truth value of test samples :return: recall(Float), the rec
metric.py:9
Methodmost_recent_sessions
(self, sessions)
baselines/KNN-based/SKNN.py:49
Methodmost_recent_sessions
(self, sessions)
baselines/KNN-based/STAN.py:68
Functionndcg_score
(y_pre, y_true, k=5)
baselines/pop.py:355
Functionndcg_score
(y_pre, y_true, k=5)
baselines/Bert4Rec.py:402
Functionndcg_score
get NDCG@k :param y_pre: numpy (batch_size,x) :param y_true: y_truth: list[batch_size][ground_truth_num] :param k: k :return: NDC
baselines/STAMP.py:339
Methodpredict
(self, session_id, session_items, session_timestamp, k=20)
baselines/KNN-based/STAN.py:131
Functionpreprocess_buys
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED)
Preprocess_rr.py:91
Functionpreprocess_days_test
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED, min_item_support=MIN_
Preprocess_rr.py:49
Functionpreprocess_days_test_min_date
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED, min_item_sup
Preprocess_rr.py:57
Functionpreprocess_info
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED, min_item_support=MIN_ITEM_SUPPORT,
Preprocess_rr.py:77
Functionpreprocess_org
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED, min_item_support=MIN_ITEM_SUPPORT,
Preprocess_rr.py:31
Functionpreprocess_org_min_date
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED, min_item_support=M
Preprocess_rr.py:39
Functionpreprocess_save
(path=DATA_PATH, file=DATA_FILE, path_proc=DATA_PATH_PROCESSED, min_item_support=MIN_ITEM_SUPPORT,
Preprocess_rr.py:83
Functionsplit_seq_train
(sid, timestamp, seq)
Preprocess_dn.py:167
Functionspop
()
baselines/pop.py:389
← previous101–162 of 162, ranked by callers