Skip to content

Latest commit

 

History

History
92 lines (67 loc) · 3.11 KB

README.md

File metadata and controls

92 lines (67 loc) · 3.11 KB

Interpretable Contextual Team-aware Item Recommendation: Application in Multiplayer Online Battle Arena Game

Open In Colab

See paper in arXiv

tir-model

We release the PyTorch code of the TTIR model.

Content

Prerequisites

The code is built with following libraries:

Dataset

The used dataset is available here.

Code

We develop this project using Google Colab. That's why you must have a Google Account and the dataset in a gDrive folder. Furthermore, you have to change these paths according to the location of the dataset.

train_path = '/content/gdrive/My Drive/Proyecto_RecSys/dataset/train_splits.pkl'
test_path = '/content/gdrive/My Drive/Proyecto_RecSys/dataset/test_splits.pkl'
champion_path = '/content/gdrive/My Drive/Proyecto_RecSys/dataset/champion_types.pkl'

And the comet parameters (api_key, project_name, workspace)

comet_logger = CometLogger(
    experiment_name=conf['exp_name'],
    api_key = 'YOUR_KEY',
    project_name="YOUR_PROJECT_NAME",
    workspace = 'YOUR_WORKSPACE'
)

Baselines

This work uses the proposed models in Data mining for item recommendation in MOBA games paper as baselines.

Results

This method outperforms the state of the art approaches and explains the result.

Method Precision@6 Recall@6 F1@6 MAP@6
TTIR 0.492 0.756 0.596 0.805
CNN 0.484 0.744 0.586 0.795
ANN 0.476 0.732 0.566 0.785

tir-att

Citation

If you find this repository useful for your research, please consider citing our paper:

@inproceedings{10.1145/3383313.3412211,
author = {Villa, Andr\'{e}s and Araujo, Vladimir and Cattan, Francisca and Parra, Denis},
title = {Interpretable Contextual Team-Aware Item Recommendation: Application in Multiplayer Online Battle Arena Games},
year = {2020},
isbn = {9781450375832},
publisher = {Association for Computing Machinery},
address = {New York, NY, USA},
url = {https://doi.org/10.1145/3383313.3412211},
doi = {10.1145/3383313.3412211},
booktitle = {Fourteenth ACM Conference on Recommender Systems},
pages = {503–508},
numpages = {6},
keywords = {Item Recommendation, Deep Learning, MOBA Games},
location = {Virtual Event, Brazil},
series = {RecSys '20}
}

For any questions, welcome to create an issue or contact Andrés Villa ([email protected]) - Vladimir Araujo ([email protected]).