MCPcopy Create free account

hub / github.com/LehengTHU/Agent4Rec / functions

Functions304 in github.com/LehengTHU/Agent4Rec

⨍ Functions 304◇ Types & classes 42
Functionap
recommenders/evaluator/backend/cpp/include/metric.h:46
Functionarg_top_k_1d
recommenders/util/cython/include/arg_topk.h:15
Functionarg_top_k_2d
recommenders/util/cython/include/arg_topk.h:29
Functionbar_group_mean
(n_groups, groups)
scripts/utils.py:84
Functionbar_rating
(x, y, y_label)
scripts/utils.py:110
Functionbatch_randint_choice
Return random integers from `0` (inclusive) to `high` (exclusive). :param high: integer :param size: 1-D array_like :param replace: bool
recommenders/util/tool.py:143
Functionchecktensor
(tensor)
recommenders/models/base/utils.py:216
Functionchoice_one
(idx)
recommenders/util/tool.py:160
Methodclear
Clear memory contents.
simulation/memory.py:61
Functionconformity
(x)
datasets/ml-1m/2_get_user_statistic.py:177
Functioncount_genres
(x)
datasets/ml-1m/2_get_user_statistic.py:93
Functioncpp_evaluate_matrix
recommenders/evaluator/backend/cpp/include/evaluate.h:45
Functioncsr_to_user_dict_bytime
(time_matrix,train_matrix)
recommenders/util/tool.py:78
Functiondecorate
(func)
recommenders/util/tool.py:173
Functiondiversity
(x)
datasets/ml-1m/2_get_user_statistic.py:127
MethodensureDir
(dir_path)
simulation/base/abstract_arena.py:74
Functioneval_one_user
recommenders/evaluator/backend/cpp/include/evaluate.h:23
Methodevaluate
Evaluate `model`. Args: model: The model need to be evaluated. This model must have a method `predict_for_eval(se
recommenders/evaluator/abstract_evaluator.py:22
Methodevaluate
Evaluate `model` in user groups. Args: model: The model need to be evaluated. This model must have a method `pred
recommenders/evaluator/grouped_evaluator.py:91
Methodevaluate
Evaluate `model`. Args: model: The model need to be evaluated. This model must have a method `predict_for_eval(se
recommenders/evaluator/backend/python/uni_evaluator.py:98
Methodevaluate
Evaluate `model`. Args: model: The model need to be evaluated. This model must have a method `predict_for_eval(se
recommenders/evaluator/backend/cpp/uni_evaluator.py:110
Functionfix_seeds
(seed=101)
simulation/utils.py:11
Methodforward
(self, users, pos_items, neg_items)
recommenders/models/MF.py:47
Methodforward
(self, users, pos_items, neg_items)
recommenders/models/LightGCN.py:49
Methodforward
(self)
recommenders/models/Pop.py:25
Methodforward
(self)
recommenders/models/Random.py:23
Methodforward
(self, users, pos_items, neg_items)
recommenders/models/InfoNCE.py:49
Methodforward
(self, users, pos_items)
recommenders/models/InfoNCE.py:87
Methodforward
(self)
recommenders/models/base/abstract_model.py:68
Functionget_completion
(prompt, sys_prompt, model="gpt-3.5-turbo", temperature=0)
datasets/ml-1m/4_generate_persona.py:292
Functionget_data_format
(data_format)
recommenders/util/tool.py:45
Methodget_not_candidate
(self)
recommenders/data.py:324
Functionget_rotation_matrix
Find the rotation matrix associated with counterclockwise rotation about the given axis by theta radians. Credit: http://stackoverflow.co
recommenders/models/base/utils.py:224
Functionhist_2D
(group_data, labels, save_name, x_label, order, text_place, is_discrete=True, bins=60)
scripts/utils.py:293
Functionhist_3D
(group_data, labels, is_discrete=True, bins=60)
scripts/utils.py:230
Functionhist_group_results
(groups, bins=20)
scripts/utils.py:72
Functionhit
(rank, ground_truth)
recommenders/evaluator/backend/python/metric.py:8
Functionhook
(grad)
recommenders/models/base/utils.py:252
Functionhr
recommenders/evaluator/backend/cpp/include/metric.h:109
Methodimportance_weighting
Weighting the importance of memory according to the results of recommendation and the personal taste
simulation/memory.py:283
Methodimportance_weighting
Weighting the importance of memory according to the results of recommendation and the personal taste
simulation/base/abstract_memory.py:19
Methodinitialize_all_avatars
initialize avatars
simulation/arena.py:40
Functionkl_divergence
(p, q)
scripts/utils.py:69
Methodload_additional_info
(self)
simulation/arena.py:24
Methodload_memory_variables
Return key-value pairs given the text input to the chain.
simulation/memory.py:65
Functionmap
(rank, ground_truth)
recommenders/evaluator/backend/python/metric.py:34
Functionmax_top_k
(a, top_k=50)
recommenders/util/tool.py:189
Methodmemory_variables
Input keys this memory class will load dynamically.
simulation/memory.py:57
Functionmerge_user_list_no_dup
(user_lists)
recommenders/models/base/utils.py:30
Methodmetrics_info
Get all metrics information. Returns: str: A string consist of all metrics information， such as `"Precision@10 Pre
recommenders/evaluator/abstract_evaluator.py:13
Methodmetrics_info
Get all metrics information. Returns: str: A string consist of all metrics information， such as `"Precision@10 Pre
recommenders/evaluator/grouped_evaluator.py:82
Methodmetrics_info
Get all metrics information. Returns: str: A string consist of all metrics information， such as `"Precision@10
recommenders/evaluator/proxy_evaluator.py:87
Methodmetrics_info
Get all metrics information. Returns: str: A string consist of all metrics information， such as `"Precision@10
recommenders/evaluator/backend/python/uni_evaluator.py:86
Methodmetrics_info
Get all metrics information. Returns: str: A string consist of all metrics information， such as `"Precision@10
recommenders/evaluator/backend/cpp/uni_evaluator.py:98
Functionmrr
(rank, ground_truth)
recommenders/evaluator/backend/python/metric.py:58
Functionmrr
recommenders/evaluator/backend/cpp/include/metric.h:86
Functionndcg
(rank, ground_truth)
recommenders/evaluator/backend/python/metric.py:47
Functionndcg
recommenders/evaluator/backend/cpp/include/metric.h:66
Functionnoise_validator
Validates the noise provided
recommenders/util/tool.py:110
Methodpage_generator
generate one page items for one avatar
simulation/base/abstract_arena.py:212
Functionpair_plot
(y1, y2)
scripts/utils.py:164
Methodparams_str
Get a summary of parameters. Returns: str: A string summary of parameters.
recommenders/util/configurator.py:104
Methodparse_response
(self, response)
simulation/arena.py:413
Methodpause_to_reflect_feeling
Reflect on recent observations and generate 'insights'.
simulation/memory.py:238
Methodpause_to_reflect_taste
Reflect on recent observations and generate 'insights'.
simulation/memory.py:226
Functionplot_bar
(data)
scripts/utils.py:214
Functionplot_rating
(x, y, type_name)
scripts/utils.py:132
Functionprecision
recommenders/evaluator/backend/cpp/include/metric.h:17
Methodpredict
(self, users, items=None)
recommenders/models/Pop.py:28
Methodpredict
(self, users, items=None)
recommenders/models/Random.py:26
Methodpredict
(self, users, items=None)
recommenders/models/MultVAE.py:161
Functionprepare_dir
(dir_path)
scripts/utils.py:62
Methodreaction_to_forced_items
Summarize the feelings of the avatar for recommended item list.
simulation/avatar.py:212
Functionrecall
(rank, ground_truth)
recommenders/evaluator/backend/python/metric.py:27
Functionrecall
recommenders/evaluator/backend/cpp/include/metric.h:31
Methodreflect
Generate a high level understanding of previous memories
simulation/base/abstract_memory.py:26
Methodreflection
Reflect on the observation bank
simulation/base/abstract_avatar.py:30
Functionrestore_best_checkpoint
Restore the best performance checkpoint
recommenders/models/base/utils.py:121
Methodrestore_checkpoint
If a checkpoint exists, restores the PyTorch model from the checkpoint. Returns the model and the current epoch.
simulation/base/abstract_arena.py:119
Methodsave_context
Save the context of this model run to memory.
simulation/memory.py:52
Functionsave_grad
(name)
recommenders/models/base/utils.py:251
Methodsave_results
save the results of the simulation
simulation/arena.py:479
Methodsave_user_dict_to_txt
(user_dict, base_path, filename)
simulation/arena.py:484
Methodscore_normalizer
(val: float)
simulation/avatar.py:71
Methodset_optimizer
(self)
recommenders/models/MultVAE.py:31
Methodsimulate_all_avatars
excute the simulation for all avatars
simulation/arena.py:188
Methodsimulate_one_avatar
excute the simulation for one avatar
simulation/base/abstract_arena.py:230
Functionsplit_grp_view
(data,grp_idx)
recommenders/models/base/utils.py:204
Methodtime_weighting
Weighting the memory according to the time
simulation/memory.py:277
Methodtime_weighting
Weighting the memory according to the time
simulation/base/abstract_memory.py:13
Functiontimer
The timer decorator
recommenders/util/tool.py:249
Methodtrain_one_epoch
(self, epoch)
recommenders/models/MF.py:15
Methodtrain_one_epoch
(self, epoch)
recommenders/models/LightGCN.py:15
Methodtrain_one_epoch
(self, epoch)
recommenders/models/Pop.py:18
Methodtrain_one_epoch
(self, epoch)
recommenders/models/Random.py:16
Methodtrain_one_epoch
(self, epoch)
recommenders/models/MultVAE.py:41
Methodtrain_one_epoch
(self, epoch)
recommenders/models/InfoNCE.py:16
Functiontypeassert
(*type_args, **type_kwargs)
recommenders/util/tool.py:172
Functionuniform_loss
(x, t=2)
recommenders/models/base/utils.py:271
Methodup_date_taste
Update the taste of the avatar
simulation/base/abstract_avatar.py:36
← previousnext →201–300 of 304, ranked by callers