/AKT

Primary LanguagePythonMIT LicenseMIT

Context-Aware Attentive Knowledge Tracing

KDD'2020: Context-Aware Attentive Knowledge Tracing (Pytorch implementation for AKT).

This is the code for the paper: Context-Aware Attentive Knowledge Tracing
Aritra Ghosh, Neil Heffernan, Andrew S. Lan To be presented at KDD 2020.

If you find this code useful in your research then please cite

@inproceedings{ghosh2020context,
  title={Context-Aware Attentive Knowledge Tracing},
  author={Ghosh, Aritra and Heffernan, Neil and Lan, Andrew S},
  booktitle={Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery \& Data Mining},
  year={2020}
}

Setups

The requiring environment is as bellow:

  • Linux
  • Python 3+
  • PyTorch 1.2.0
  • Scikit-learn 0.21.3
  • Scipy 1.3.1
  • Numpy 1.17.2

Running AKT.

Here are some examples for using AKT-Rasch model (on ASSISTments2009 and ASSISTments2017 datasets):

python main.py --dataset assist2009_pid --model akt_pid 
python main.py --dataset assist2017_pid --model akt_pid

Here is an example for using AKT-NonRasch model (on all datasets):

python3 main.py --dataset assist2015 --model akt_cid

Errata

Thanks to Rui Zou and Xuhao Li for pointing out a bug in line 306 of akt.py. It has been fixed now.

Contact: Aritra Ghosh (aritraghosh.iem@gmail.com).