syunnmo / KCMN

KCMN

The manuscript has not been published. This project is the Pytorch implementation for KCMN.

Dataset

In 'data' folder, we have provided the processed datasets. If you would like to access the raw datasets, the raw datasets are placed in the following links:

• ASSISTments2009 : address
• ASSISTments2012 : address
• slepemapy.cz. : address
• EdNet : address

Setups

Service:

• Linux operation system

Environment:

• python 3+
• sklearn 0.21.3
• tqdm 4.54.1
• torch 1.7.0
• numpy 1.19.2

Running KCMN

Here is a example for using FGKT model (on EdNet):

  python main.py --dataset ednet  

Explanation of parameters:

• gpu : Specify the GPU to be used, e.g '0,1,2,3'. If CPU is used then fill in -1.
• exercise_embed_dim : Number of exercise embedding dimensions.
• batch_size : Number of batch size.
• max_step : The allowed maximum length of a sequence.
• init_lr : Learning rate.
• num_heads : Number of head attentions.
• mode : Selection of integration function.