main()
Experiment
Experiment.__init__()
Experiment.run()
ExperimentResult
ExperimentResult.__init__()
prepare_minimal_dataset()
prepare_simple_pattern()
prepare_synthetic_data()
aggregate_user_history()
create_dataset_unbiased()
create_temporal_splits()
create_user_splits()
stable_join()
Dataset
Dataset.__init__()
Dataset.user_in_test
DatasetBase
DatasetBase.__init__()
evaluate_assigned()
evaluate_item_rec()
evaluate_mtch()
evaluate_user_rec()
Dual
Dual.__init__()
assign_mtch()
comm_mat()
setup_linprog()
triu_to_mat()
LinProg
LinProg.__init__()
EMA
EMA.__init__()
Pop
Pop.__init__()
Pop.transform()
Rand
Rand.__init__()
Rand.transform()
loss()
HawkesPoisson
HawkesPoisson.__init__()
RNN
RNN.__init__()
RNNModel
RNNModel.__init__()
RNNModel.add_module()
RNNModel.apply()
RNNModel.bfloat16()
RNNModel.buffers()
RNNModel.children()
RNNModel.cpu()
RNNModel.cuda()
RNNModel.double()
RNNModel.eval()
RNNModel.extra_repr()
RNNModel.float()
RNNModel.forward()
RNNModel.get_buffer()
RNNModel.get_extra_state()
RNNModel.get_parameter()
RNNModel.get_submodule()
RNNModel.half()
RNNModel.ipu()
RNNModel.load_state_dict()
RNNModel.modules()
RNNModel.named_buffers()
RNNModel.named_children()
RNNModel.named_modules()
RNNModel.named_parameters()
RNNModel.parameters()
RNNModel.register_backward_hook()
RNNModel.register_buffer()
RNNModel.register_forward_hook()
RNNModel.register_forward_pre_hook()
RNNModel.register_full_backward_hook()
RNNModel.register_load_state_dict_post_hook()
RNNModel.register_module()
RNNModel.register_parameter()
RNNModel.requires_grad_()
RNNModel.set_extra_state()
RNNModel.share_memory()
RNNModel.state_dict()
RNNModel.to()
RNNModel.to_empty()
RNNModel.train()
RNNModel.type()
RNNModel.xpu()
RNNModel.zero_grad()
TransformerModel
TransformerModel.__init__()
TransformerModel.add_module()
TransformerModel.apply()
TransformerModel.bfloat16()
TransformerModel.buffers()
TransformerModel.children()
TransformerModel.cpu()
TransformerModel.cuda()
TransformerModel.double()
TransformerModel.eval()
TransformerModel.extra_repr()
TransformerModel.float()
TransformerModel.forward()
TransformerModel.get_buffer()
TransformerModel.get_extra_state()
TransformerModel.get_parameter()
TransformerModel.get_submodule()
TransformerModel.half()
TransformerModel.ipu()
TransformerModel.load_state_dict()
TransformerModel.modules()
TransformerModel.named_buffers()
TransformerModel.named_children()
TransformerModel.named_modules()
TransformerModel.named_parameters()
TransformerModel.parameters()
TransformerModel.register_backward_hook()
TransformerModel.register_buffer()
TransformerModel.register_forward_hook()
TransformerModel.register_forward_pre_hook()
TransformerModel.register_full_backward_hook()
TransformerModel.register_load_state_dict_post_hook()
TransformerModel.register_module()
TransformerModel.register_parameter()
TransformerModel.requires_grad_()
TransformerModel.set_extra_state()
TransformerModel.share_memory()
TransformerModel.state_dict()
TransformerModel.to()
TransformerModel.to_empty()
TransformerModel.train()
TransformerModel.type()
TransformerModel.xpu()
TransformerModel.zero_grad()
PositionalEncoding
PositionalEncoding.__init__()
PositionalEncoding.add_module()
PositionalEncoding.apply()
PositionalEncoding.bfloat16()
PositionalEncoding.buffers()
PositionalEncoding.children()
PositionalEncoding.cpu()
PositionalEncoding.cuda()
PositionalEncoding.double()
PositionalEncoding.eval()
PositionalEncoding.extra_repr()
PositionalEncoding.float()
PositionalEncoding.forward()
PositionalEncoding.get_buffer()
PositionalEncoding.get_extra_state()
PositionalEncoding.get_parameter()
PositionalEncoding.get_submodule()
PositionalEncoding.half()
PositionalEncoding.ipu()
PositionalEncoding.load_state_dict()
PositionalEncoding.modules()
PositionalEncoding.named_buffers()
PositionalEncoding.named_children()
PositionalEncoding.named_modules()
PositionalEncoding.named_parameters()
PositionalEncoding.parameters()
PositionalEncoding.register_backward_hook()
PositionalEncoding.register_buffer()
PositionalEncoding.register_forward_hook()
PositionalEncoding.register_forward_pre_hook()
PositionalEncoding.register_full_backward_hook()
PositionalEncoding.register_load_state_dict_post_hook()
PositionalEncoding.register_module()
PositionalEncoding.register_parameter()
PositionalEncoding.requires_grad_()
PositionalEncoding.set_extra_state()
PositionalEncoding.share_memory()
PositionalEncoding.state_dict()
PositionalEncoding.to()
PositionalEncoding.to_empty()
PositionalEncoding.train()
PositionalEncoding.type()
PositionalEncoding.xpu()
PositionalEncoding.zero_grad()
Transformer
Transformer.__init__()
BayesLM
BayesLM.__init__()
BayesLM.transform()
ItemKNN
ItemKNN.__init__()
TF_IDF
TF_IDF.__init__()
assign_topk()
default_random_split()
empty_cache_on_exit()
explode_user_titles()
export_jsondump()
extract_past_ij()
extract_user_item()
fill_factory_inplace()
filter_min_len()
groupby_unexplode()
indices2csr()
perplexity()
sample_groupA()
warn_nan_output()
MissingModel
MissingModel.__init__()
timed
timed.__init__()
dual_clip()
dual_complete()
dual_iterate()
dual_solve_u()
dual_v()
grad_u()
lagrangian()
primal_solution()
plot_mtch_results()
plot_rec_results()
auto_cast_lazy_score()
auto_device()
auto_tensor()
batch_op_iter()
find_iloc()
get_batch_size()
matrix_reindex()
score_op()
sps_to_torch()
ElementWiseExpression
ElementWiseExpression.__init__()
ElementWiseExpression.collate_fn()
LazyDenseMatrix
LazyDenseMatrix.__init__()
LazyDenseMatrix.collate_fn()
LazyExpressionBase
LazyExpressionBase.__init__()
LazyScoreBase
LazyScoreBase.__init__()
LazyScoreBase.collate_fn()
LazyScoreModel
LazyScoreModel.__init__()
LazySparseMatrix
LazySparseMatrix.__init__()
LazySparseMatrix.collate_fn()
MatMulExpression
MatMulExpression.__init__()
MatMulExpression.collate_fn()
RandScore
RandScore.__init__()
RandScore.collate_fn()
VAEExpression
VAEExpression.__init__()
VAEExpression.collate_fn()
Classes
RNNModel(rnn_type, ntoken, ninp, nhid, nlayers)
TransformerModel(*args[, tie_weights])
Modules
rime.models.third_party.word_language_model.model