├── .idea ├── .gitignore ├── MIRRN-main.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml └── modules.xml ├── CITATION ├── LICENSE ├── README.md ├── data ├── alipay │ └── preprocess.py ├── taobao │ └── preprocess.py └── tmall │ └── preprocess.py ├── fuxictr ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── features.cpython-39.pyc │ ├── metrics.cpython-39.pyc │ ├── utils.cpython-39.pyc │ └── version.cpython-39.pyc ├── autotuner.py ├── datasets │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── avazu.cpython-39.pyc │ │ ├── criteo.cpython-39.pyc │ │ └── kkbox.cpython-39.pyc │ ├── avazu.py │ ├── criteo.py │ └── kkbox.py ├── features.py ├── metrics.py ├── preprocess │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── build_dataset.cpython-39.pyc │ │ ├── feature_processor.cpython-39.pyc │ │ ├── normalizer.cpython-39.pyc │ │ └── tokenizer.cpython-39.pyc │ ├── build_dataset.py │ ├── feature_processor.py │ ├── normalizer.py │ └── tokenizer.py ├── pytorch │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ └── torch_utils.cpython-39.pyc │ ├── dataloaders │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── npz_block_dataloader.cpython-39.pyc │ │ │ ├── npz_dataloader.cpython-39.pyc │ │ │ ├── parquet_block_dataloader.cpython-39.pyc │ │ │ ├── parquet_dataloader.cpython-39.pyc │ │ │ └── rank_dataloader.cpython-39.pyc │ │ ├── npz_block_dataloader.py │ │ ├── npz_dataloader.py │ │ ├── parquet_block_dataloader.py │ │ ├── parquet_dataloader.py │ │ └── rank_dataloader.py │ ├── layers │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── activations.cpython-39.pyc │ │ │ └── pooling.cpython-39.pyc │ │ ├── activations.py │ │ ├── attentions │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-39.pyc │ │ │ │ ├── dot_product_attention.cpython-39.pyc │ │ │ │ ├── squeeze_excitation.cpython-39.pyc │ │ │ │ └── target_attention.cpython-39.pyc │ │ │ ├── dot_product_attention.py │ │ │ ├── squeeze_excitation.py │ │ │ └── target_attention.py │ │ ├── blocks │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-39.pyc │ │ │ │ ├── factorization_machine.cpython-39.pyc │ │ │ │ ├── logistic_regression.cpython-39.pyc │ │ │ │ └── mlp_block.cpython-39.pyc │ │ │ ├── factorization_machine.py │ │ │ ├── logistic_regression.py │ │ │ └── mlp_block.py │ │ ├── embeddings │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-39.pyc │ │ │ │ ├── feature_embedding.cpython-39.pyc │ │ │ │ └── pretrained_embedding.cpython-39.pyc │ │ │ ├── feature_embedding.py │ │ │ └── pretrained_embedding.py │ │ ├── interactions │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-39.pyc │ │ │ │ ├── bilinear_interaction.cpython-39.pyc │ │ │ │ ├── compressed_interaction_net.cpython-39.pyc │ │ │ │ ├── cross_net.cpython-39.pyc │ │ │ │ ├── holographic_interaction.cpython-39.pyc │ │ │ │ ├── inner_product.cpython-39.pyc │ │ │ │ └── interaction_machine.cpython-39.pyc │ │ │ ├── bilinear_interaction.py │ │ │ ├── compressed_interaction_net.py │ │ │ ├── cross_net.py │ │ │ ├── holographic_interaction.py │ │ │ ├── inner_product.py │ │ │ └── interaction_machine.py │ │ └── pooling.py │ ├── models │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── multitask_model.cpython-39.pyc │ │ │ └── rank_model.cpython-39.pyc │ │ ├── multitask_model.py │ │ └── rank_model.py │ └── torch_utils.py ├── tensorflow │ ├── __init__.py │ ├── dataloaders │ │ ├── __init__.py │ │ └── tf_dataloader.py │ ├── layers │ │ ├── __init__.py │ │ ├── blocks │ │ │ ├── __init__.py │ │ │ ├── factorization_machine.py │ │ │ ├── linear.py │ │ │ ├── logistic_regression.py │ │ │ └── mlp_block.py │ │ ├── embeddings │ │ │ ├── __init__.py │ │ │ └── feature_embedding.py │ │ ├── interactions │ │ │ ├── __init__.py │ │ │ ├── cross_net.py │ │ │ └── inner_product.py │ │ └── pooling.py │ ├── models │ │ ├── __init__.py │ │ └── rank_model.py │ └── tf_utils.py ├── utils.py └── version.py ├── main ├── build_alipay_to_parquet.py ├── build_taobao_to_parquet.py ├── build_tmall_to_parquet.py ├── config │ ├── General_config │ │ ├── dataset_config.yaml │ │ └── model_config.yaml │ ├── build_alipay_to_parquet_config │ │ └── dataset_config.yaml │ ├── build_taobao_to_parquet_config │ │ └── dataset_config.yaml │ └── build_tmall_to_parquet_config │ │ └── dataset_config.yaml └── run_expid.py └── model_zoo ├── MIRRN └── src │ ├── MIRRN.py │ ├── __init__.py │ └── __pycache__ │ ├── MIRRN.cpython-39.pyc │ └── __init__.cpython-39.pyc ├── __init__.py └── __pycache__ └── __init__.cpython-39.pyc /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/MIRRN-main.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/.idea/MIRRN-main.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /CITATION: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/CITATION -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/README.md -------------------------------------------------------------------------------- /data/alipay/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/data/alipay/preprocess.py -------------------------------------------------------------------------------- /data/taobao/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/data/taobao/preprocess.py -------------------------------------------------------------------------------- /data/tmall/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/data/tmall/preprocess.py -------------------------------------------------------------------------------- /fuxictr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/__init__.py -------------------------------------------------------------------------------- /fuxictr/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/__pycache__/features.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/__pycache__/features.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/__pycache__/metrics.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/__pycache__/metrics.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/__pycache__/version.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/__pycache__/version.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/autotuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/autotuner.py -------------------------------------------------------------------------------- /fuxictr/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/__init__.py -------------------------------------------------------------------------------- /fuxictr/datasets/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/datasets/__pycache__/avazu.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/__pycache__/avazu.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/datasets/__pycache__/criteo.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/__pycache__/criteo.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/datasets/__pycache__/kkbox.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/__pycache__/kkbox.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/datasets/avazu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/avazu.py -------------------------------------------------------------------------------- /fuxictr/datasets/criteo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/criteo.py -------------------------------------------------------------------------------- /fuxictr/datasets/kkbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/datasets/kkbox.py -------------------------------------------------------------------------------- /fuxictr/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/features.py -------------------------------------------------------------------------------- /fuxictr/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/metrics.py -------------------------------------------------------------------------------- /fuxictr/preprocess/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/__init__.py -------------------------------------------------------------------------------- /fuxictr/preprocess/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/preprocess/__pycache__/build_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/__pycache__/build_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/preprocess/__pycache__/feature_processor.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/__pycache__/feature_processor.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/preprocess/__pycache__/normalizer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/__pycache__/normalizer.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/preprocess/__pycache__/tokenizer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/__pycache__/tokenizer.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/preprocess/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/build_dataset.py -------------------------------------------------------------------------------- /fuxictr/preprocess/feature_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/feature_processor.py -------------------------------------------------------------------------------- /fuxictr/preprocess/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/normalizer.py -------------------------------------------------------------------------------- /fuxictr/preprocess/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/preprocess/tokenizer.py -------------------------------------------------------------------------------- /fuxictr/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fuxictr/pytorch/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/__pycache__/torch_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/__pycache__/torch_utils.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__pycache__/npz_block_dataloader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__pycache__/npz_block_dataloader.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__pycache__/npz_dataloader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__pycache__/npz_dataloader.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__pycache__/parquet_block_dataloader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__pycache__/parquet_block_dataloader.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__pycache__/parquet_dataloader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__pycache__/parquet_dataloader.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/__pycache__/rank_dataloader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/__pycache__/rank_dataloader.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/npz_block_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/npz_block_dataloader.py -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/npz_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/npz_dataloader.py -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/parquet_block_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/parquet_block_dataloader.py -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/parquet_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/parquet_dataloader.py -------------------------------------------------------------------------------- /fuxictr/pytorch/dataloaders/rank_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/dataloaders/rank_dataloader.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/__pycache__/activations.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/__pycache__/activations.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/__pycache__/pooling.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/__pycache__/pooling.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/activations.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/__pycache__/dot_product_attention.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/__pycache__/dot_product_attention.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/__pycache__/squeeze_excitation.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/__pycache__/squeeze_excitation.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/__pycache__/target_attention.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/__pycache__/target_attention.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/dot_product_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/dot_product_attention.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/squeeze_excitation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/squeeze_excitation.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/attentions/target_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/attentions/target_attention.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/__pycache__/factorization_machine.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/__pycache__/factorization_machine.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/__pycache__/logistic_regression.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/__pycache__/logistic_regression.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/__pycache__/mlp_block.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/__pycache__/mlp_block.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/factorization_machine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/factorization_machine.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/logistic_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/logistic_regression.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/blocks/mlp_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/blocks/mlp_block.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/embeddings/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/embeddings/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/embeddings/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/embeddings/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/embeddings/__pycache__/feature_embedding.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/embeddings/__pycache__/feature_embedding.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/embeddings/__pycache__/pretrained_embedding.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/embeddings/__pycache__/pretrained_embedding.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/embeddings/feature_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/embeddings/feature_embedding.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/embeddings/pretrained_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/embeddings/pretrained_embedding.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/bilinear_interaction.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/bilinear_interaction.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/compressed_interaction_net.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/compressed_interaction_net.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/cross_net.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/cross_net.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/holographic_interaction.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/holographic_interaction.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/inner_product.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/inner_product.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/__pycache__/interaction_machine.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/__pycache__/interaction_machine.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/bilinear_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/bilinear_interaction.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/compressed_interaction_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/compressed_interaction_net.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/cross_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/cross_net.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/holographic_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/holographic_interaction.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/inner_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/inner_product.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/interactions/interaction_machine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/interactions/interaction_machine.py -------------------------------------------------------------------------------- /fuxictr/pytorch/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/layers/pooling.py -------------------------------------------------------------------------------- /fuxictr/pytorch/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/models/__init__.py -------------------------------------------------------------------------------- /fuxictr/pytorch/models/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/models/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/models/__pycache__/multitask_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/models/__pycache__/multitask_model.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/models/__pycache__/rank_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/models/__pycache__/rank_model.cpython-39.pyc -------------------------------------------------------------------------------- /fuxictr/pytorch/models/multitask_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/models/multitask_model.py -------------------------------------------------------------------------------- /fuxictr/pytorch/models/rank_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/models/rank_model.py -------------------------------------------------------------------------------- /fuxictr/pytorch/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/pytorch/torch_utils.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fuxictr/tensorflow/dataloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/dataloaders/__init__.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/dataloaders/tf_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/dataloaders/tf_dataloader.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/__init__.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/blocks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/blocks/__init__.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/blocks/factorization_machine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/blocks/factorization_machine.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/blocks/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/blocks/linear.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/blocks/logistic_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/blocks/logistic_regression.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/blocks/mlp_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/blocks/mlp_block.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/embeddings/__init__.py: -------------------------------------------------------------------------------- 1 | from .feature_embedding import * 2 | 3 | -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/embeddings/feature_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/embeddings/feature_embedding.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/interactions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/interactions/__init__.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/interactions/cross_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/interactions/cross_net.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/interactions/inner_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/interactions/inner_product.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/layers/pooling.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .rank_model import BaseModel -------------------------------------------------------------------------------- /fuxictr/tensorflow/models/rank_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/models/rank_model.py -------------------------------------------------------------------------------- /fuxictr/tensorflow/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/tensorflow/tf_utils.py -------------------------------------------------------------------------------- /fuxictr/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/fuxictr/utils.py -------------------------------------------------------------------------------- /fuxictr/version.py: -------------------------------------------------------------------------------- 1 | __version__="2.3.1" 2 | -------------------------------------------------------------------------------- /main/build_alipay_to_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/build_alipay_to_parquet.py -------------------------------------------------------------------------------- /main/build_taobao_to_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/build_taobao_to_parquet.py -------------------------------------------------------------------------------- /main/build_tmall_to_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/build_tmall_to_parquet.py -------------------------------------------------------------------------------- /main/config/General_config/dataset_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/config/General_config/dataset_config.yaml -------------------------------------------------------------------------------- /main/config/General_config/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/config/General_config/model_config.yaml -------------------------------------------------------------------------------- /main/config/build_alipay_to_parquet_config/dataset_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/config/build_alipay_to_parquet_config/dataset_config.yaml -------------------------------------------------------------------------------- /main/config/build_taobao_to_parquet_config/dataset_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/config/build_taobao_to_parquet_config/dataset_config.yaml -------------------------------------------------------------------------------- /main/config/build_tmall_to_parquet_config/dataset_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/config/build_tmall_to_parquet_config/dataset_config.yaml -------------------------------------------------------------------------------- /main/run_expid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/main/run_expid.py -------------------------------------------------------------------------------- /model_zoo/MIRRN/src/MIRRN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/model_zoo/MIRRN/src/MIRRN.py -------------------------------------------------------------------------------- /model_zoo/MIRRN/src/__init__.py: -------------------------------------------------------------------------------- 1 | from .MIRRN import * 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /model_zoo/MIRRN/src/__pycache__/MIRRN.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/model_zoo/MIRRN/src/__pycache__/MIRRN.cpython-39.pyc -------------------------------------------------------------------------------- /model_zoo/MIRRN/src/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/model_zoo/MIRRN/src/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /model_zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/model_zoo/__init__.py -------------------------------------------------------------------------------- /model_zoo/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTC-StarTeam/MIRRN/HEAD/model_zoo/__pycache__/__init__.cpython-39.pyc --------------------------------------------------------------------------------