Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AkaliKong/MiniOneRec
/ functions
Functions
426 in github.com/AkaliKong/MiniOneRec
⨍
Functions
426
◇
Types & classes
56
↳
Endpoints
24
Method
forward
(self, states, len_states)
sasrec.py:105
Method
forward
(self, states, len_states)
sasrec.py:241
Method
forward
(self, x)
rq/generate_indices_plus.py:20
Method
forward
(self, x)
rq/rqkmeans_plus.py:21
Method
forward
(self, x, use_sk=True)
rq/models/vq.py:63
Method
forward
(self, x, use_sk=True)
rq/models/rq.py:39
Method
forward
(self, input_feature)
rq/models/layers.py:42
Method
forward
(self, x, use_sk=True)
rq/models/rqvae.py:61
Method
forward_eval
(self, states, len_states)
sasrec.py:190
Method
forward_eval
(self, states, len_states)
sasrec.py:259
Function
gao
(path, item_path)
calc.py:12
Function
gao
(category, metadata=None, reviews=None, K=5, st_year=2017, st_month=10, ed_year=2018, ed_month=11, output=True
data/process.py:16
Method
generate_example_prompt
(self, data_point)
data.py:128
Method
generate_example_prompt
(self, data_point)
data.py:282
Method
generate_example_prompt
(self, data_point)
data.py:604
Method
generate_prompt_description
(self, history)
ts_rec_data.py:538
Method
generate_prompt_description
(self, history)
data.py:1232
Method
get_all
(self)
ts_rec_data.py:133
Method
get_codebook
(self)
rq/models/vq.py:29
Method
get_codebook
(self)
rq/models/rq.py:32
Method
get_codebook_entry
(self, indices, shape=None)
rq/models/vq.py:32
Function
get_cosine_schedule_with_warmup
( optimizer, num_warmup_steps, num_training_steps, num_cycles: float = 0.5, last_epoch: int = -1 )
sft.py:76
Function
get_cosine_schedule_with_warmup
( optimizer, num_warmup_steps, num_training_steps, num_cycles: float = 0.5, last_epoch: int = -1 )
ts_rec_sft.py:71
Function
get_cosine_schedule_with_warmup
( optimizer, num_warmup_steps, num_training_steps, num_cycles: float = 0.5, last_epoch: int = -1 )
sft_gpr.py:76
Method
get_history
(self, row)
data.py:494
Method
get_inputs_list
(self)
ts_rec_data.py:139
Method
get_inputs_list
(self)
ts_rec_data.py:273
Method
get_inputs_list
(self)
ts_rec_data.py:413
Method
get_inputs_list
(self)
ts_rec_data.py:645
Method
get_inputs_list
(self)
data.py:68
Function
get_res_batch
(model_name, prompt_list, max_tokens, api_info)
rq/text2emb/utils.py:22
Function
hepo_reward
(prompts, completions)
rl_gpr.py:94
Method
init_weights
(self, module)
rq/models/layers.py:35
Function
load_pickle
(filename)
rq/text2emb/utils.py:331
Function
load_plm
(model_path='bert-base-uncased')
rq/text2emb/utils.py:288
Function
main
( base_model: str = "", train_file: str = "", info_file: str = "", category: str = "", tes
evaluate.py:38
Function
make_inters_in_order
(inters)
rq/text2emb/utils.py:336
Function
mask_grad
(grad)
sft.py:172
Function
mask_grad
(grad)
ts_rec_sft.py:201
Function
mask_grad
(grad)
sft_gpr.py:226
Function
merge
(input_path, output_path, cuda_list)
merge.py:6
Function
ndcg_rule_reward
(prompts, completions)
rl_gpr.py:199
Function
ndcg_rule_reward
(prompts, completions)
rl.py:160
Function
normalize
Applies layer normalization. Args: inputs: A tensor with 2 or more dimensions, where the first dimension has `batch_size`. ep
utility.py:46
Function
pad_history
(itemlist,length,pad_item)
utility.py:22
Method
pre
(self, idx)
data.py:154
Method
pre
(self, idx)
data.py:242
Method
pre
(self, idx)
data.py:305
Method
pre
(self, idx)
data.py:381
Method
pre
(self, idx)
data.py:421
Method
pre
(self, idx)
data.py:512
Method
pre
(self, idx)
data.py:629
Method
pre
(self, idx)
data.py:746
Method
pre
(self, idx)
data.py:875
Method
pre
(self, idx)
data.py:946
Method
pre
(self, idx)
data.py:1024
Method
pre
(self, idx)
data.py:1105
Method
pre
(self, idx)
data.py:1277
Method
pre
(self, idx)
data.py:1404
Method
pre
(self, idx)
data.py:1584
Method
pre
(self, idx)
data.py:1765
Method
prediction_step
(self, model, inputs, prediction_loss_only, ignore_keys: Optional[list[str]] = None)
minionerec_trainer.py:1075
Method
prefix_allowed_tokens_fn
(self, batch_id, input_ids)
minionerec_trainer.py:588
Function
prefix_allowed_tokens_fn_semantic
(batch_id, input_ids)
evaluate.py:122
Function
prefix_allowed_tokens_fn_title
(batch_id, input_ids)
evaluate.py:128
Method
reg_loss
(self, reg_weights=1e-2)
utility.py:176
Function
rule_reward
(prompts, completions)
rl_gpr.py:225
Function
rule_reward
(prompts, completions)
rl.py:186
Method
select_completion
(completions, target)
minionerec_trainer.py:820
Function
semantic_reward
(prompts, completions)
rl_gpr.py:238
Function
semantic_reward
(prompts, completions)
rl.py:199
Method
setUp
(self)
tests/test_minimax_provider.py:341
Method
setUpClass
(cls)
data_test.py:37
Function
set_device
(gpu_id)
rq/text2emb/utils.py:281
Function
split
(input_path, output_path, cuda_list)
split.py:5
Method
tearDownClass
(cls)
data_test.py:102
Method
test_D3Dataset_initialization
Test D3Dataset initialization
data_test.py:119
Method
test_EvalD3Dataset_initialization
Test EvalD3Dataset initialization
data_test.py:131
Method
test_EvalSidDataset_initialization
Test SidItemFeatDataset initialization
data_test.py:195
Method
test_FusionSeqRecDataset_initialization
Test FusionSeqRecDataset initialization
data_test.py:243
Method
test_PreferenceSFTDataset_initialization
Test PreferenceSFTDataset initialization
data_test.py:271
Method
test_RLSid2TitleDataset_initialization
Test RLSid2TitleDataset initialization
data_test.py:220
Method
test_RLSidhis2TitleDataset_initialization
Test RLSidhis2TitleDataset initialization
data_test.py:231
Method
test_RLTitle2SidDataset_initialization
Test RLTitle2SidDataset initialization
data_test.py:209
Method
test_SFTData_initialization
Test SFTData initialization
data_test.py:106
Method
test_SidDataset_initialization
Test SidDataset initialization
data_test.py:144
Method
test_SidItemFeatDataset_initialization
Test SidItemFeatDataset initialization
data_test.py:182
Method
test_SidSFTDataset_initialization
Test SidSFTDataset initialization
data_test.py:156
Method
test_TitleHistory2SidSFTDataset_initialization
Test TitleHistory2SidSFTDataset initialization
data_test.py:257
Method
test_UserPreference2sidSFTDataset_initialization
Test UserPreference2sidSFTDataset initialization
data_test.py:284
Method
test_authorization_header
(self, mock_post)
tests/test_minimax_provider.py:208
Method
test_batch_completion
(self)
tests/test_minimax_provider.py:357
Method
test_custom_base_url
(self, mock_post)
tests/test_minimax_provider.py:95
Method
test_deepseek_provider
(self, mock_ds)
tests/test_minimax_provider.py:36
Method
test_default_base_url
(self, mock_post)
tests/test_minimax_provider.py:107
Method
test_default_provider_is_openai
(self, mock_openai)
tests/test_minimax_provider.py:29
Method
test_default_temperature
(self, mock_post)
tests/test_minimax_provider.py:290
Method
test_empty_prompt_list
(self, mock_post)
tests/test_minimax_provider.py:115
Method
test_end_to_end_preference_prompt
Test with a prompt similar to actual MiniOneRec usage.
tests/test_minimax_provider.py:373
Method
test_exception_retries
(self, mock_post)
tests/test_minimax_provider.py:219
← previous
next →
301–400 of 426, ranked by callers