Training & Model | Comment |
---|---|
riiid_pytorch_simple.ipynb | Simplified code of Eric's (MPWARE) Transformer in Pytorch - basic model |
Training & Model | Comment |
---|---|
demo_riiid_train.ipynb | File used for training the TF transformer model in Colab |
demo_riiid_transformer.ipynb | The functional TF transformer model with masking |
Pre-processing | Comment |
---|---|
demo_riiid_preprocessing.ipynb | Pre-processing of the features for the model |