MCPcopy Create free account

hub / github.com/AkaliKong/MiniOneRec / types & classes

Types & classes56 in github.com/AkaliKong/MiniOneRec

↓ 5 callersClassEmbDataset
rq/datasets.py:6
↓ 5 callersClassRQVAE
rq/models/rqvae.py:10
↓ 5 callersClassSidDataset
data.py:354
↓ 5 callersClassSidSFTDataset
data.py:397
↓ 4 callersClassTokenizer
ts_rec_data.py:14
↓ 3 callersClassFusionSeqRecDataset
data.py:1126
↓ 3 callersClassRLTitle2SidDataset
data.py:789
↓ 3 callersClassSASRec
sasrec.py:214
↓ 3 callersClassSFTData
data.py:108
↓ 3 callersClassSidItemFeatDataset
data.py:678
↓ 2 callersClassConstrainedLogitsProcessor
LogitProcessor.py:24
↓ 2 callersClassEvalSidDataset
data.py:597
↓ 2 callersClassMLPLayers
rq/models/layers.py:7
↓ 2 callersClassRLSeqTitle2SidDataset
data.py:890
↓ 2 callersClassReReTrainer
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 callersClassRepeatRandomSampler
Sampler that repeats the indices of a dataset N times. Args: data_source (`Sized`): Dataset to sample from. repe
minionerec_trainer.py:80
↓ 2 callersClassSidSFTDataset
ts_rec_data.py:38
↓ 2 callersClassTitleHistory2SidSFTDataset
data.py:1332
↓ 2 callersClassTrainer
rq/trainer.py:14
↓ 2 callersClassUserPreference2sidSFTDataset
data.py:1629
↓ 1 callersClassCaser
sasrec.py:127
↓ 1 callersClassD3Dataset
data.py:202
↓ 1 callersClassEvalD3Dataset
data.py:262
↓ 1 callersClassFusionSeqRecDataset
ts_rec_data.py:423
↓ 1 callersClassGRU
sasrec.py:86
↓ 1 callersClassMockTokenizer
data_test.py:15
↓ 1 callersClassMultiHeadAttention
SASRecModules_ori.py:26
↓ 1 callersClassPositionwiseFeedForward
SASRecModules_ori.py:7
↓ 1 callersClassPreferenceSFTDataset
data.py:1448
↓ 1 callersClassRLSid2TitleDataset
data.py:969
↓ 1 callersClassRLSidhis2TitleDataset
data.py:1039
↓ 1 callersClassRecDataset
sasrec.py:414
↓ 1 callersClassResidualEncoderWrapper
rq/generate_indices_plus.py:15
↓ 1 callersClassResidualEncoderWrapper
rq/rqkmeans_plus.py:15
↓ 1 callersClassResidualVectorQuantizer
References: SoundStream: An End-to-End Neural Audio Codec https://arxiv.org/pdf/2107.03312.pdf
rq/models/rq.py:7
↓ 1 callersClassSidItemFeatDataset
ts_rec_data.py:283
↓ 1 callersClassSidTokenFeatDataset
ts_rec_data.py:147
↓ 1 callersClassTokenExtender
sft.py:30
↓ 1 callersClassTokenExtender
ts_rec_sft.py:26
↓ 1 callersClassTokenExtender
sft_gpr.py:30
↓ 1 callersClassTokenizer
data.py:13
↓ 1 callersClassVAFT_Trainer
sft_gpr.py:90
↓ 1 callersClassVectorQuantizer
rq/models/vq.py:7
ClassBaseDataset
data.py:37
ClassCSVBaseDataset
data.py:87
ClassJSONBaseDataset
data.py:97
ClassMemoryUnit
utility.py:152
ClassNeuProcessEncoder
utility.py:104
ClassSidSFTDataset_GPR
data.py:469
ClassTestDataModule
data_test.py:35
ClassTestGetResBatchDispatch
Test that get_res_batch correctly dispatches to the right provider.
tests/test_minimax_provider.py:25
ClassTestMiniMaxBatch
Test get_minimax_batch function.
tests/test_minimax_provider.py:57
ClassTestMiniMaxIntegration
Integration tests that call real MiniMax API (skipped without API key).
tests/test_minimax_provider.py:338
ClassTestSingleMiniMaxRequest
Test _single_minimax_request function.
tests/test_minimax_provider.py:122
ClassTestTemperatureClamping
Test MiniMax temperature clamping behavior.
tests/test_minimax_provider.py:254
ClassTestTestGenerationConfigHasNoTopKTopP
Ensure test_generation_config disables top_k/top_p to avoid no-valid-token warnings.
tests/test_minimax_provider.py:396