MCPcopy Create free account

hub / github.com/AkaliKong/MiniOneRec / functions

Functions426 in github.com/AkaliKong/MiniOneRec

⨍ Functions 426◇ Types & classes 56↳ Endpoints 24
↓ 1 callersFunctioncost_fun
(X)
rq/rqkmeans_faiss.py:143
↓ 1 callersFunctioncreate_item_features_amazon18_style
Create item features like amazon18_data_process
data/amazon18_data_process_gpr.py:390
↓ 1 callersFunctioncreate_item_features_amazon18_style
Create item features like amazon18_data_process
data/amazon18_data_process.py:382
↓ 1 callersFunctioncreate_item_info_file
Create item info file (sid -> title -> item_id mapping)
convert_dataset_gpr.py:42
↓ 1 callersFunctioncreate_item_info_file
Create item info file (sid -> title -> item_id mapping)
convert_dataset.py:42
↓ 1 callersFunctioncreate_minimal_csv
Helper to create minimal CSV files for testing
data_test.py:30
↓ 1 callersFunctiondeal_with_deduplicate
Handle duplicates by appending row index
rq/rqkmeans_constrained.py:131
↓ 1 callersFunctiondeal_with_deduplicate
(df)
rq/generate_indices_plus.py:23
↓ 1 callersMethodemb_encode
(self, input_tensor)
utility.py:124
↓ 1 callersFunctionencode_with_rq
(rq, data, codebook_size, verbose=True)
rq/rqkmeans_faiss.py:69
↓ 1 callersFunctionensure_dir
(dir_path)
rq/utils.py:6
↓ 1 callersFunctionestimate_tau
(residuals, centroids, sample_size=4000, percentile=90, min_tau=1e-6)
rq/rqkmeans_faiss.py:105
↓ 1 callersFunctionevaluate
( encodings, num_beams=10, max_new_tokens=64, length_penalty=1
evaluate.py:152
↓ 1 callersMethodforward
(self, states, len_states)
sasrec.py:166
↓ 1 callersMethodgenerate_formatted_prompt
(self, prompt, response)
ts_rec_data.py:577
↓ 1 callersMethodgenerate_formatted_prompt
(self, prompt, response)
data.py:940
↓ 1 callersMethodgenerate_formatted_prompt
(self, prompt, response)
data.py:1271
↓ 1 callersFunctiongenerate_interaction_list_json2csv_style
Generate interaction list like json2csv for 8:1:1 split
data/amazon18_data_process_gpr.py:218
↓ 1 callersFunctiongenerate_interaction_list_json2csv_style
Generate interaction list like json2csv for 8:1:1 split
data/amazon18_data_process.py:218
↓ 1 callersFunctiongenerate_item_embedding
(args, item_text_list, tokenizer, model, accelerator, word_drop_ratio=-1)
rq/text2emb/amazon_text2emb.py:52
↓ 1 callersFunctiongenerate_item_embedding
(args, item_text_list, tokenizer, model, accelerator, word_drop_ratio=-1)
rq/text2emb/amazon_text2emb_gpr.py:52
↓ 1 callersMethodgenerate_prompt
(self, data_point)
ts_rec_data.py:210
↓ 1 callersMethodgenerate_prompt
(self, data_point)
ts_rec_data.py:352
↓ 1 callersMethodgenerate_prompt
(self, data_point)
data.py:733
↓ 1 callersMethodgenerate_prompt
(self, data_point)
data.py:862
↓ 1 callersMethodgenerate_prompt
(self, inter_titles)
data.py:910
↓ 1 callersMethodgenerate_prompt
(self, data_point)
data.py:1015
↓ 1 callersMethodgenerate_prompt_title
(self, history)
ts_rec_data.py:535
↓ 1 callersMethodgenerate_prompt_title
(self, history)
data.py:1229
↓ 1 callersFunctiongenerate_sids
(args)
rq/generate_indices_plus.py:83
↓ 1 callersFunctiongenerate_text
(item2feature, features)
rq/text2emb/amazon_text2emb.py:21
↓ 1 callersFunctiongenerate_text
(item2feature, features)
rq/text2emb/amazon_text2emb_gpr.py:21
↓ 1 callersMethodget_all
(self)
data.py:62
↓ 1 callersFunctionget_collision_item
(all_indices_str)
rq/generate_indices.py:29
↓ 1 callersFunctionget_collision_item
(all_indices_str)
rq/models/generate_indices.py:29
↓ 1 callersFunctionget_deepseek_batch
(model_name, prompt_list, max_tokens, api_info)
rq/text2emb/utils.py:85
↓ 1 callersMethodget_history
(self, row)
ts_rec_data.py:541
↓ 1 callersMethodget_history
(self, row)
data.py:77
↓ 1 callersMethodget_history
(self, row)
data.py:135
↓ 1 callersMethodget_history
(self, row)
data.py:223
↓ 1 callersMethodget_history
(self, row)
data.py:288
↓ 1 callersMethodget_history
(self, row)
data.py:362
↓ 1 callersMethodget_history
(self, row)
data.py:403
↓ 1 callersMethodget_history
(self, row)
data.py:611
↓ 1 callersMethodget_history
(self, row)
data.py:913
↓ 1 callersMethodget_history
(self, row)
data.py:1077
↓ 1 callersMethodget_history
(self, row)
data.py:1235
↓ 1 callersMethodget_history
Extract user history from title sequence and target semantic ID
data.py:1371
↓ 1 callersMethodget_history
Extract and format user history and preference
data.py:1557
↓ 1 callersFunctionget_indices_count
(all_indices_str)
rq/generate_indices.py:23
↓ 1 callersFunctionget_indices_count
(all_indices_str)
rq/models/generate_indices.py:23
↓ 1 callersMethodget_input_and_target
Extract and format user history, preference, and target item
data.py:1740
↓ 1 callersMethodget_inputs
(self)
ts_rec_data.py:267
↓ 1 callersFunctionget_local_time
r"""Get current time Returns: str: current time
rq/utils.py:24
↓ 1 callersFunctionget_minimax_batch
(model_name, prompt_list, max_tokens, api_info)
rq/text2emb/utils.py:178
↓ 1 callersMethodget_new_tokens
(self)
sft.py:42
↓ 1 callersMethodget_new_tokens
(self)
ts_rec_sft.py:38
↓ 1 callersMethodget_new_tokens
(self)
sft_gpr.py:42
↓ 1 callersFunctionget_openai_batch
(model_name, prompt_list, max_tokens, api_info)
rq/text2emb/utils.py:34
↓ 1 callersMethodinit_emb
(self, data)
rq/models/vq.py:40
↓ 1 callersFunctioninitialize_sid_token_with_text
(model, tokenizer, new_token_id, keywords)
ts_rec_sft.py:84
↓ 1 callersFunctionk_core_filter_amazon23
K-core filtering for Amazon23 reviews. Uses: - user_id - asin - timestamp (converted from sort_timestamp) And beh
data/amazon23_data_process.py:143
↓ 1 callersFunctionk_core_filtering_json2csv_style
Perform k-core filtering using json2csv style logic
data/amazon18_data_process_gpr.py:121
↓ 1 callersFunctionk_core_filtering_json2csv_style
Perform k-core filtering using json2csv style logic
data/amazon18_data_process.py:121
↓ 1 callersFunctionkmeans
( samples, num_clusters, num_iters = 10, )
rq/models/layers.py:69
↓ 1 callersFunctionload_data
(args)
rq/text2emb/amazon_text2emb.py:14
↓ 1 callersFunctionload_data
(args)
rq/text2emb/amazon_text2emb_gpr.py:14
↓ 1 callersFunctionload_dataset
Load all dataset files
convert_dataset_gpr.py:13
↓ 1 callersFunctionload_dataset
Load all dataset files
convert_dataset.py:13
↓ 1 callersFunctionload_metadata_amazon23
(metadata_file)
data/amazon23_data_process.py:94
↓ 1 callersFunctionload_metadata_json2csv_style
Load metadata using json2csv style processing
data/amazon18_data_process_gpr.py:63
↓ 1 callersFunctionload_metadata_json2csv_style
Load metadata using json2csv style processing
data/amazon18_data_process.py:63
↓ 1 callersFunctionload_model
(args, dim)
rq/generate_indices_plus.py:40
↓ 1 callersFunctionload_qwen_model
(model_path)
rq/text2emb/amazon_text2emb.py:148
↓ 1 callersFunctionload_qwen_model
(model_path)
rq/text2emb/amazon_text2emb_gpr.py:155
↓ 1 callersFunctionload_review_data_amazon18_style
Load review data like amazon18_data_process
data/amazon18_data_process_gpr.py:352
↓ 1 callersFunctionload_review_data_amazon18_style
Load review data like amazon18_data_process
data/amazon18_data_process.py:344
↓ 1 callersFunctionload_reviews_amazon23
(reviews_file)
data/amazon23_data_process.py:48
↓ 1 callersFunctionload_reviews_json2csv_style
Load reviews using json2csv style processing
data/amazon18_data_process_gpr.py:96
↓ 1 callersFunctionload_reviews_json2csv_style
Load reviews using json2csv style processing
data/amazon18_data_process.py:96
↓ 1 callersFunctionmain
()
convert_dataset_gpr.py:167
↓ 1 callersFunctionmain
()
convert_dataset.py:161
↓ 1 callersFunctionmain
(topk, data_file_train, data_file_test, data_file_valid)
sasrec.py:428
↓ 1 callersFunctionmain
()
data/amazon23_data_process.py:516
↓ 1 callersFunctionmain
()
rq/rqkmeans_faiss.py:235
↓ 1 callersFunctionparse_args
()
sasrec.py:25
↓ 1 callersFunctionparse_args
()
data/amazon18_data_process_gpr.py:534
↓ 1 callersFunctionparse_args
()
data/amazon23_data_process.py:490
↓ 1 callersFunctionparse_args
()
data/amazon18_data_process.py:467
↓ 1 callersFunctionparse_args
()
rq/rqkmeans_constrained.py:168
↓ 1 callersFunctionparse_args
()
rq/rqvae.py:14
↓ 1 callersFunctionparse_args
()
rq/generate_indices_plus.py:169
↓ 1 callersFunctionparse_args
()
rq/rqkmeans_plus.py:90
↓ 1 callersFunctionparse_args
()
rq/text2emb/amazon_text2emb.py:159
↓ 1 callersFunctionparse_args
()
rq/text2emb/amazon_text2emb_gpr.py:166
↓ 1 callersMethodpre
(self, idx)
ts_rec_data.py:79
↓ 1 callersMethodpre
(self, idx)
data.py:74
↓ 1 callersFunctionpreprocess_text
(args)
rq/text2emb/amazon_text2emb.py:45
↓ 1 callersFunctionpreprocess_text
(args)
rq/text2emb/amazon_text2emb_gpr.py:45
↓ 1 callersFunctionprocess_dataset_recursive
Process dataset with recursive year reduction like json2csv
data/amazon18_data_process_gpr.py:501
← previousnext →101–200 of 426, ranked by callers