MCPcopy Create free account

hub / github.com/LehengTHU/Agent4Rec / functions

Functions304 in github.com/LehengTHU/Agent4Rec

↓ 24 callersMethodwrite_log
(self, log, color=None, attrs=None, print=False)
simulation/avatar.py:315
↓ 8 callersFunctionmerge_user_list
(user_lists)
recommenders/models/base/utils.py:20
↓ 7 callersMethodadd_memory
Add an observation or memory to the agent's memory bank.
simulation/memory.py:244
↓ 5 callersMethod_flush
(self)
recommenders/util/logger.py:48
↓ 5 callersMethodpredict
(self, users, items=None)
recommenders/models/MF.py:71
↓ 5 callersMethodtrain
(self)
recommenders/models/base/abstract_RS.py:115
↓ 4 callersMethod_reaction
Summarize the feelings of the avatar for recommended item list.
simulation/avatar.py:90
↓ 4 callersMethodcompute
(self)
recommenders/models/base/abstract_model.py:48
↓ 4 callersMethodformat_memories_detail
(self, relevant_memories: List[Document])
simulation/memory.py:84
↓ 4 callersFunctionhelper_load
(filename)
recommenders/data.py:20
↓ 4 callersFunctionrandint_choice
Return random integers from `0` (inclusive) to `high` (exclusive).
recommenders/util/tool.py:126
↓ 3 callersMethod__init__
(self)
recommenders/util/data_iterator.py:15
↓ 3 callersMethod_format_memory_detail
(self, memory: Document, prefix: str = "")
simulation/memory.py:90
↓ 3 callersFunctionargmax_top_k
(a, top_k=50)
recommenders/util/tool.py:194
↓ 3 callersFunctionclear_checkpoint
(checkpoint_dir)
recommenders/models/base/utils.py:142
↓ 3 callersFunctiondescribe_interactions
(df)
datasets/ml-1m/1_get_cf_data.py:21
↓ 3 callersFunctionensureDir
(dir_path)
recommenders/util/tool.py:11
↓ 3 callersMethodfetch_memories
Fetch related memories.
simulation/memory.py:76
↓ 3 callersMethodget_completion
(self, prompt, llm="gpt-3.5-turbo", temperature=0)
simulation/memory.py:108
↓ 3 callersFunctionint_to_user_dict
convert a list of interactions into a dictionary that maps each user to a list of their interactions input: df with columns ['user_id',
datasets/ml-1m/1_get_cf_data.py:29
↓ 3 callersFunctionsave_user_dict_to_txt
(user_dict, base_path, filename)
datasets/ml-1m/1_get_cf_data.py:46
↓ 2 callersMethod__init__
(self, args, data)
recommenders/models/Pop.py:22
↓ 2 callersMethod__init__
(self, args, data)
recommenders/models/MultVAE.py:89
↓ 2 callersMethod__init__
(self, args, data)
recommenders/models/InfoNCE.py:45
↓ 2 callersFunction_get_hours_passed
Get the hours passed between two datetime objects.
simulation/retriever.py:11
↓ 2 callersMethod_read_config_file
(self, filename)
recommenders/util/configurator.py:81
↓ 2 callersMethodenqueue
recommenders/util/cython/include/thread_pool.h:67
↓ 2 callersMethodeval_score_matrix
(self, score_matrix, test_items, metric, top_k, thread_num)
recommenders/evaluator/backend/python/uni_evaluator.py:155
↓ 2 callersMethodevaluate
Evaluate `model`. Args: model: The model need to be evaluated. This model must have a method `predict_for_eval(se
recommenders/evaluator/proxy_evaluator.py:96
↓ 2 callersFunctionevaluation
(args, data, model, epoch, base_path, evaluator, name="valid")
recommenders/models/base/utils.py:150
↓ 2 callersFunctionfix_seeds
(seed=101)
scripts/utils.py:52
↓ 2 callersFunctionget_precision
(y_true, y_pred)
simulation/utils.py:27
↓ 2 callersFunctionget_recall
(y_true, y_pred)
simulation/utils.py:24
↓ 2 callersFunctionnaive_sparse2tensor
(data)
recommenders/models/MultVAE.py:21
↓ 2 callersMethodpage_generator
generate one page items for one avatar
simulation/arena.py:67
↓ 2 callersMethodpredict
(self, users, items=None)
recommenders/models/base/abstract_model.py:72
↓ 1 callersMethod__init__
(self, args, data)
recommenders/models/MF.py:44
↓ 1 callersMethod__init__
(self, args, data)
recommenders/models/LightGCN.py:46
↓ 1 callersMethod__init__
(self, args, data)
recommenders/models/Random.py:20
↓ 1 callersMethod__str__
(self)
recommenders/util/configurator.py:151
↓ 1 callersMethod_convert_sp_mat_to_sp_tensor
(self, X)
recommenders/data.py:274
↓ 1 callersMethod_get_combined_score_list
Return the combined score for a document.
simulation/retriever.py:66
↓ 1 callersMethod_read_cmd_arg
(self)
recommenders/util/configurator.py:70
↓ 1 callersMethod_user_feeling_reflection
Return the user's feeling about recent movies.
simulation/memory.py:205
↓ 1 callersMethod_user_satisfaction_reflection
Return the user's feeling about recent movies.
simulation/memory.py:185
↓ 1 callersMethod_user_taste_reflection
Return the user's taste about recent movies.
simulation/memory.py:166
↓ 1 callersMethodadd_documents
Add documents to vectorstore.
simulation/retriever.py:151
↓ 1 callersMethodadd_special_model_attr
(self, args)
recommenders/data.py:144
↓ 1 callersMethodasync_simulate_one_avatar
async excute the simulation for one avatar avatar_id: the id of the simulated avatar
simulation/arena.py:216
↓ 1 callersMethodasync_validate_one_avatar
async validate the effectiveness of the model for one avatar avatar_id: the id of the simulated avatar
simulation/arena.py:111
↓ 1 callersMethodcritical
(self, message)
recommenders/util/logger.py:68
↓ 1 callersFunctioncsr_to_user_dict
convert a scipy.sparse.csr_matrix to a dict, where the key is row number, and value is the non-empty index in each row.
recommenders/util/tool.py:66
↓ 1 callersMethoddebug
(self, message)
recommenders/util/logger.py:52
↓ 1 callersMethoddecode
(self, z)
recommenders/models/MultVAE.py:130
↓ 1 callersMethoddisplay_item_with_adver
(self, store_path, avatar_id, i, id_on_page, movies_on_page)
simulation/arena.py:461
↓ 1 callersMethoddisplay_only_adver_item
(self, store_path, avatar_id, i, id_on_page, movies_on_page)
simulation/arena.py:428
↓ 1 callersMethoddocument_hyper_params_results
(self, base_path, n_rets)
recommenders/models/base/abstract_RS.py:180
↓ 1 callersMethoddocument_running_loss
(self, losses:list, epoch, t_one_epoch, prefix="")
recommenders/models/base/abstract_RS.py:173
↓ 1 callersMethodencode
(self, input)
recommenders/models/MultVAE.py:109
↓ 1 callersFunctionensureDir
(dir_path)
recommenders/models/base/utils.py:198
↓ 1 callersMethoderror
(self, message)
recommenders/util/logger.py:64
↓ 1 callersMethodeval_and_check_early_stop
(self, epoch)
recommenders/models/base/abstract_RS.py:272
↓ 1 callersMethodexcute
The whole process of the simulation
simulation/base/abstract_arena.py:36
↓ 1 callersMethodexecute
(self)
recommenders/models/base/abstract_RS.py:76
↓ 1 callersFunctionfix_seeds
(seed=101)
datasets/ml-1m/2_get_user_statistic.py:11
↓ 1 callersFunctionfix_seeds
(seed=101)
datasets/ml-1m/3_get_movie_detail.py:11
↓ 1 callersFunctionfix_seeds
(seed=101)
datasets/ml-1m/4_generate_persona.py:11
↓ 1 callersFunctionfix_seeds
(seed=101)
datasets/ml-1m/1_get_cf_data.py:11
↓ 1 callersFunctionfix_seeds
(seed=101)
recommenders/models/base/utils.py:258
↓ 1 callersMethodformat_memories_simple
(self, relevant_memories: List[Document])
simulation/memory.py:94
↓ 1 callersMethodforward
(self, input)
recommenders/models/MultVAE.py:104
↓ 1 callersFunctiongenerate_init_info
(s)
datasets/ml-1m/5_document_persona.py:8
↓ 1 callersMethodgetSparseGraph
(self)
recommenders/data.py:282
↓ 1 callersFunctionget_accuracy
(y_true, y_pred)
simulation/utils.py:21
↓ 1 callersMethodget_dataloader
(self)
recommenders/data.py:247
↓ 1 callersMethodget_evaluators
(self, data, pop_mask=None)
recommenders/models/base/abstract_RS.py:386
↓ 1 callersFunctionget_f1
(y_true, y_pred)
simulation/utils.py:30
↓ 1 callersMethodget_full_rankings
document the full rankings of the items, according to a specific cf model
simulation/base/abstract_arena.py:150
↓ 1 callersMethodget_relevant_documents
Return documents that are relevant to the query.
simulation/retriever.py:108
↓ 1 callersMethodget_salient_docs
Return documents that are salient to the query.
simulation/retriever.py:94
↓ 1 callersMethodget_weight
(self)
recommenders/data.py:253
↓ 1 callersFunctionhelper_load_train
(filename)
scripts/utils.py:13
↓ 1 callersFunctionhelper_load_train
(filename)
datasets/ml-1m/4_generate_persona.py:21
↓ 1 callersFunctionhelper_load_train
(filename)
recommenders/data.py:39
↓ 1 callersMethodinfo
(self, message)
recommenders/util/logger.py:56
↓ 1 callersMethodinit_embedding
(self)
recommenders/models/base/abstract_model.py:41
↓ 1 callersMethodinit_memory
Initialize the memory of the avatar
simulation/avatar.py:66
↓ 1 callersMethodinit_weights
(self)
recommenders/models/MultVAE.py:138
↓ 1 callersMethodinitialize_all_avatars
initialize all avatars
simulation/base/abstract_arena.py:202
↓ 1 callersMethodload_additional_info
load additional information for the simulation
simulation/base/abstract_arena.py:242
↓ 1 callersMethodload_data
(self)
recommenders/data.py:148
↓ 1 callersMethodload_data
load the data for simulation
simulation/base/abstract_arena.py:87
↓ 1 callersMethodload_recommender
load the recommender for simulation
simulation/base/abstract_arena.py:106
↓ 1 callersMethodload_saved_args
load the recommender args, which is saved when training the recommender
simulation/base/abstract_arena.py:53
↓ 1 callersMethodloss_function
(self, recon_x, x, mu, logvar, anneal=1.0)
recommenders/models/MultVAE.py:34
↓ 1 callersMethodmake_next_decision
(self, remember=False, current_page=None)
simulation/avatar.py:148
↓ 1 callersMethodmodify_saveID
(self)
recommenders/models/base/abstract_RS.py:167
↓ 1 callersFunctionpad_sequences
Pads sequences to the same length. Args: sequences (list): A list of lists, where each element is a sequence. value (int or float
recommenders/util/tool.py:199
↓ 1 callersFunctionparse_args
()
parse.py:3
↓ 1 callersFunctionparse_args
()
recommenders/parse.py:4
next →1–100 of 304, ranked by callers