├── LICENSE ├── README.md └── script ├── common ├── Dice.py ├── __init__.py ├── rnn.py ├── tgin.py └── utils.py ├── dataset_process ├── __init__.py ├── content_wise │ ├── __init__.py │ ├── generate_voc.py │ ├── local_aggregator.py │ ├── prepare_data_cw.sh │ ├── process_data.py │ └── split_by_user.py ├── data_iterator.py ├── gen_wnd_edges.py ├── prepare_data.py ├── read_content_wise_dataset_zhibo.py └── shuffle.py ├── models ├── DEI2N.py ├── DEI2N_MHTA.py ├── DEI2N_NO_IL.py ├── DEI2N_NO_TIM.py ├── DEI2N_NO_UIM.py ├── DIAN.py ├── DIHN.py ├── DIN.py ├── DIN_V2_Gru_Gru_att.py ├── DIN_V2_Gru_QA_attGru.py ├── DIN_V2_Gru_Vec_attGru.py ├── DIN_V2_Gru_Vec_attGru_Neg.py ├── DIN_V2_Gru_att_Gru.py ├── DNN.py ├── DNN_Multi_Head.py ├── LR.py ├── PNN.py ├── WideDeep.py ├── __init__.py └── base_model.py ├── settings.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/README.md -------------------------------------------------------------------------------- /script/common/Dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/common/Dice.py -------------------------------------------------------------------------------- /script/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/common/__init__.py -------------------------------------------------------------------------------- /script/common/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/common/rnn.py -------------------------------------------------------------------------------- /script/common/tgin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/common/tgin.py -------------------------------------------------------------------------------- /script/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/common/utils.py -------------------------------------------------------------------------------- /script/dataset_process/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/__init__.py -------------------------------------------------------------------------------- /script/dataset_process/content_wise/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/content_wise/__init__.py -------------------------------------------------------------------------------- /script/dataset_process/content_wise/generate_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/content_wise/generate_voc.py -------------------------------------------------------------------------------- /script/dataset_process/content_wise/local_aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/content_wise/local_aggregator.py -------------------------------------------------------------------------------- /script/dataset_process/content_wise/prepare_data_cw.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/content_wise/prepare_data_cw.sh -------------------------------------------------------------------------------- /script/dataset_process/content_wise/process_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/content_wise/process_data.py -------------------------------------------------------------------------------- /script/dataset_process/content_wise/split_by_user.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/content_wise/split_by_user.py -------------------------------------------------------------------------------- /script/dataset_process/data_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/data_iterator.py -------------------------------------------------------------------------------- /script/dataset_process/gen_wnd_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/gen_wnd_edges.py -------------------------------------------------------------------------------- /script/dataset_process/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/prepare_data.py -------------------------------------------------------------------------------- /script/dataset_process/read_content_wise_dataset_zhibo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/read_content_wise_dataset_zhibo.py -------------------------------------------------------------------------------- /script/dataset_process/shuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/dataset_process/shuffle.py -------------------------------------------------------------------------------- /script/models/DEI2N.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DEI2N.py -------------------------------------------------------------------------------- /script/models/DEI2N_MHTA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DEI2N_MHTA.py -------------------------------------------------------------------------------- /script/models/DEI2N_NO_IL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DEI2N_NO_IL.py -------------------------------------------------------------------------------- /script/models/DEI2N_NO_TIM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DEI2N_NO_TIM.py -------------------------------------------------------------------------------- /script/models/DEI2N_NO_UIM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DEI2N_NO_UIM.py -------------------------------------------------------------------------------- /script/models/DIAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIAN.py -------------------------------------------------------------------------------- /script/models/DIHN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIHN.py -------------------------------------------------------------------------------- /script/models/DIN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIN.py -------------------------------------------------------------------------------- /script/models/DIN_V2_Gru_Gru_att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIN_V2_Gru_Gru_att.py -------------------------------------------------------------------------------- /script/models/DIN_V2_Gru_QA_attGru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIN_V2_Gru_QA_attGru.py -------------------------------------------------------------------------------- /script/models/DIN_V2_Gru_Vec_attGru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIN_V2_Gru_Vec_attGru.py -------------------------------------------------------------------------------- /script/models/DIN_V2_Gru_Vec_attGru_Neg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIN_V2_Gru_Vec_attGru_Neg.py -------------------------------------------------------------------------------- /script/models/DIN_V2_Gru_att_Gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DIN_V2_Gru_att_Gru.py -------------------------------------------------------------------------------- /script/models/DNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DNN.py -------------------------------------------------------------------------------- /script/models/DNN_Multi_Head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/DNN_Multi_Head.py -------------------------------------------------------------------------------- /script/models/LR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/LR.py -------------------------------------------------------------------------------- /script/models/PNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/PNN.py -------------------------------------------------------------------------------- /script/models/WideDeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/WideDeep.py -------------------------------------------------------------------------------- /script/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/__init__.py -------------------------------------------------------------------------------- /script/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/models/base_model.py -------------------------------------------------------------------------------- /script/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/settings.py -------------------------------------------------------------------------------- /script/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mengxiaozhibo/DEI2N/HEAD/script/train.py --------------------------------------------------------------------------------