Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AkaliKong/MiniOneRec
/ types & classes
Types & classes
56 in github.com/AkaliKong/MiniOneRec
⨍
Functions
426
◇
Types & classes
56
↳
Endpoints
24
↓ 5 callers
Class
EmbDataset
rq/datasets.py:6
↓ 5 callers
Class
RQVAE
rq/models/rqvae.py:10
↓ 5 callers
Class
SidDataset
data.py:354
↓ 5 callers
Class
SidSFTDataset
data.py:397
↓ 4 callers
Class
Tokenizer
ts_rec_data.py:14
↓ 3 callers
Class
FusionSeqRecDataset
data.py:1126
↓ 3 callers
Class
RLTitle2SidDataset
data.py:789
↓ 3 callers
Class
SASRec
sasrec.py:214
↓ 3 callers
Class
SFTData
data.py:108
↓ 3 callers
Class
SidItemFeatDataset
data.py:678
↓ 2 callers
Class
ConstrainedLogitsProcessor
LogitProcessor.py:24
↓ 2 callers
Class
EvalSidDataset
data.py:597
↓ 2 callers
Class
MLPLayers
rq/models/layers.py:7
↓ 2 callers
Class
RLSeqTitle2SidDataset
data.py:890
↓ 2 callers
Class
ReReTrainer
Trainer for the Group Relative Policy Optimization (GRPO) method adapted to recommendation. This algorithm was initially proposed in the pape
minionerec_trainer.py:121
↓ 2 callers
Class
RepeatRandomSampler
Sampler that repeats the indices of a dataset N times. Args: data_source (`Sized`): Dataset to sample from. repe
minionerec_trainer.py:80
↓ 2 callers
Class
SidSFTDataset
ts_rec_data.py:38
↓ 2 callers
Class
TitleHistory2SidSFTDataset
data.py:1332
↓ 2 callers
Class
Trainer
rq/trainer.py:14
↓ 2 callers
Class
UserPreference2sidSFTDataset
data.py:1629
↓ 1 callers
Class
Caser
sasrec.py:127
↓ 1 callers
Class
D3Dataset
data.py:202
↓ 1 callers
Class
EvalD3Dataset
data.py:262
↓ 1 callers
Class
FusionSeqRecDataset
ts_rec_data.py:423
↓ 1 callers
Class
GRU
sasrec.py:86
↓ 1 callers
Class
MockTokenizer
data_test.py:15
↓ 1 callers
Class
MultiHeadAttention
SASRecModules_ori.py:26
↓ 1 callers
Class
PositionwiseFeedForward
SASRecModules_ori.py:7
↓ 1 callers
Class
PreferenceSFTDataset
data.py:1448
↓ 1 callers
Class
RLSid2TitleDataset
data.py:969
↓ 1 callers
Class
RLSidhis2TitleDataset
data.py:1039
↓ 1 callers
Class
RecDataset
sasrec.py:414
↓ 1 callers
Class
ResidualEncoderWrapper
rq/generate_indices_plus.py:15
↓ 1 callers
Class
ResidualEncoderWrapper
rq/rqkmeans_plus.py:15
↓ 1 callers
Class
ResidualVectorQuantizer
References: SoundStream: An End-to-End Neural Audio Codec https://arxiv.org/pdf/2107.03312.pdf
rq/models/rq.py:7
↓ 1 callers
Class
SidItemFeatDataset
ts_rec_data.py:283
↓ 1 callers
Class
SidTokenFeatDataset
ts_rec_data.py:147
↓ 1 callers
Class
TokenExtender
sft.py:30
↓ 1 callers
Class
TokenExtender
ts_rec_sft.py:26
↓ 1 callers
Class
TokenExtender
sft_gpr.py:30
↓ 1 callers
Class
Tokenizer
data.py:13
↓ 1 callers
Class
VAFT_Trainer
sft_gpr.py:90
↓ 1 callers
Class
VectorQuantizer
rq/models/vq.py:7
Class
BaseDataset
data.py:37
Class
CSVBaseDataset
data.py:87
Class
JSONBaseDataset
data.py:97
Class
MemoryUnit
utility.py:152
Class
NeuProcessEncoder
utility.py:104
Class
SidSFTDataset_GPR
data.py:469
Class
TestDataModule
data_test.py:35
Class
TestGetResBatchDispatch
Test that get_res_batch correctly dispatches to the right provider.
tests/test_minimax_provider.py:25
Class
TestMiniMaxBatch
Test get_minimax_batch function.
tests/test_minimax_provider.py:57
Class
TestMiniMaxIntegration
Integration tests that call real MiniMax API (skipped without API key).
tests/test_minimax_provider.py:338
Class
TestSingleMiniMaxRequest
Test _single_minimax_request function.
tests/test_minimax_provider.py:122
Class
TestTemperatureClamping
Test MiniMax temperature clamping behavior.
tests/test_minimax_provider.py:254
Class
TestTestGenerationConfigHasNoTopKTopP
Ensure test_generation_config disables top_k/top_p to avoid no-valid-token warnings.
tests/test_minimax_provider.py:396