├── .github └── workflows │ ├── main.yml │ └── python-test.yml ├── .gitignore ├── CHANGE.txt ├── LICENSE ├── Makefile ├── README.md ├── XKT ├── DKT │ ├── DKT.py │ ├── __init__.py │ └── etl.py ├── DKVMN │ ├── DKVMN.py │ ├── __init__.py │ ├── etl.py │ └── net.py ├── GKT │ ├── MGKT.py │ ├── __init__.py │ ├── etl.py │ └── net.py ├── SKT │ ├── MSKT.py │ ├── __init__.py │ ├── etl.py │ ├── net.py │ └── utils.py ├── __init__.py ├── meta.py └── utils │ ├── __init__.py │ ├── etl.py │ ├── loss.py │ ├── nn │ ├── __init__.py │ └── rnn.py │ └── tests.py ├── docs └── DISCUSSION.md ├── examples ├── DKT │ ├── DKT.ipynb │ ├── DKT.py │ └── prepare_dataset.ipynb ├── DKVMN │ ├── DKVMN.ipynb │ ├── DKVMN.py │ └── prepare_dataset.ipynb ├── GKT │ ├── MGKT.py │ └── prepare_dataset.ipynb └── SKT │ ├── MSKT.py │ └── prepare_dataset.ipynb ├── pytest.ini ├── scripts ├── DKT │ ├── DKT.py │ ├── README.md │ ├── config.yml │ └── search_space.json ├── DKVMN │ ├── DKVMN.py │ ├── README.md │ ├── config.yml │ └── search_space.json ├── GKT │ ├── MGKT.py │ ├── README.md │ ├── config.yml │ └── search_space.json └── SKT │ ├── MSKT.py │ ├── README.md │ ├── config.yml │ └── search_space.json ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── dkt ├── __init__.py ├── conftest.py └── test_dkt.py ├── dkvmn ├── __init__.py ├── conftest.py └── test_dkvmn.py ├── gkt ├── __init__.py ├── conftest.py └── test_mgkt.py └── skt ├── __init__.py ├── conftest.py └── test_mskt.py /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.github/workflows/python-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/.github/workflows/python-test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/CHANGE.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/README.md -------------------------------------------------------------------------------- /XKT/DKT/DKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKT/DKT.py -------------------------------------------------------------------------------- /XKT/DKT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKT/__init__.py -------------------------------------------------------------------------------- /XKT/DKT/etl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKT/etl.py -------------------------------------------------------------------------------- /XKT/DKVMN/DKVMN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKVMN/DKVMN.py -------------------------------------------------------------------------------- /XKT/DKVMN/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKVMN/__init__.py -------------------------------------------------------------------------------- /XKT/DKVMN/etl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKVMN/etl.py -------------------------------------------------------------------------------- /XKT/DKVMN/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/DKVMN/net.py -------------------------------------------------------------------------------- /XKT/GKT/MGKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/GKT/MGKT.py -------------------------------------------------------------------------------- /XKT/GKT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/GKT/__init__.py -------------------------------------------------------------------------------- /XKT/GKT/etl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/GKT/etl.py -------------------------------------------------------------------------------- /XKT/GKT/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/GKT/net.py -------------------------------------------------------------------------------- /XKT/SKT/MSKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/SKT/MSKT.py -------------------------------------------------------------------------------- /XKT/SKT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/SKT/__init__.py -------------------------------------------------------------------------------- /XKT/SKT/etl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/SKT/etl.py -------------------------------------------------------------------------------- /XKT/SKT/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/SKT/net.py -------------------------------------------------------------------------------- /XKT/SKT/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/SKT/utils.py -------------------------------------------------------------------------------- /XKT/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/__init__.py -------------------------------------------------------------------------------- /XKT/meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/meta.py -------------------------------------------------------------------------------- /XKT/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/utils/__init__.py -------------------------------------------------------------------------------- /XKT/utils/etl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/utils/etl.py -------------------------------------------------------------------------------- /XKT/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/utils/loss.py -------------------------------------------------------------------------------- /XKT/utils/nn/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | # 2021/8/23 @ tongshiwei 3 | 4 | from .rnn import * 5 | -------------------------------------------------------------------------------- /XKT/utils/nn/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/utils/nn/rnn.py -------------------------------------------------------------------------------- /XKT/utils/tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/XKT/utils/tests.py -------------------------------------------------------------------------------- /docs/DISCUSSION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/docs/DISCUSSION.md -------------------------------------------------------------------------------- /examples/DKT/DKT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/DKT/DKT.ipynb -------------------------------------------------------------------------------- /examples/DKT/DKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/DKT/DKT.py -------------------------------------------------------------------------------- /examples/DKT/prepare_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/DKT/prepare_dataset.ipynb -------------------------------------------------------------------------------- /examples/DKVMN/DKVMN.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/DKVMN/DKVMN.ipynb -------------------------------------------------------------------------------- /examples/DKVMN/DKVMN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/DKVMN/DKVMN.py -------------------------------------------------------------------------------- /examples/DKVMN/prepare_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/DKVMN/prepare_dataset.ipynb -------------------------------------------------------------------------------- /examples/GKT/MGKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/GKT/MGKT.py -------------------------------------------------------------------------------- /examples/GKT/prepare_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/GKT/prepare_dataset.ipynb -------------------------------------------------------------------------------- /examples/SKT/MSKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/SKT/MSKT.py -------------------------------------------------------------------------------- /examples/SKT/prepare_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/examples/SKT/prepare_dataset.ipynb -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/pytest.ini -------------------------------------------------------------------------------- /scripts/DKT/DKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKT/DKT.py -------------------------------------------------------------------------------- /scripts/DKT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKT/README.md -------------------------------------------------------------------------------- /scripts/DKT/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKT/config.yml -------------------------------------------------------------------------------- /scripts/DKT/search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKT/search_space.json -------------------------------------------------------------------------------- /scripts/DKVMN/DKVMN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKVMN/DKVMN.py -------------------------------------------------------------------------------- /scripts/DKVMN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKVMN/README.md -------------------------------------------------------------------------------- /scripts/DKVMN/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKVMN/config.yml -------------------------------------------------------------------------------- /scripts/DKVMN/search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/DKVMN/search_space.json -------------------------------------------------------------------------------- /scripts/GKT/MGKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/GKT/MGKT.py -------------------------------------------------------------------------------- /scripts/GKT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/GKT/README.md -------------------------------------------------------------------------------- /scripts/GKT/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/GKT/config.yml -------------------------------------------------------------------------------- /scripts/GKT/search_space.json: -------------------------------------------------------------------------------- 1 | { 2 | "hyper_params_update:hidden_num": {"_type":"choice","_value":[16, 32, 64]} 3 | } -------------------------------------------------------------------------------- /scripts/SKT/MSKT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/SKT/MSKT.py -------------------------------------------------------------------------------- /scripts/SKT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/SKT/README.md -------------------------------------------------------------------------------- /scripts/SKT/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/scripts/SKT/config.yml -------------------------------------------------------------------------------- /scripts/SKT/search_space.json: -------------------------------------------------------------------------------- 1 | { 2 | "hyper_params_update:hidden_num": {"_type":"choice","_value":[16, 32, 64]} 3 | } -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | # create by tongshiwei on 2019-7-13 3 | -------------------------------------------------------------------------------- /tests/dkt/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | # 2021/8/24 @ tongshiwei 3 | -------------------------------------------------------------------------------- /tests/dkt/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/dkt/conftest.py -------------------------------------------------------------------------------- /tests/dkt/test_dkt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/dkt/test_dkt.py -------------------------------------------------------------------------------- /tests/dkvmn/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | # 2021/8/24 @ tongshiwei 3 | -------------------------------------------------------------------------------- /tests/dkvmn/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/dkvmn/conftest.py -------------------------------------------------------------------------------- /tests/dkvmn/test_dkvmn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/dkvmn/test_dkvmn.py -------------------------------------------------------------------------------- /tests/gkt/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | # 2021/8/24 @ tongshiwei 3 | -------------------------------------------------------------------------------- /tests/gkt/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/gkt/conftest.py -------------------------------------------------------------------------------- /tests/gkt/test_mgkt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/gkt/test_mgkt.py -------------------------------------------------------------------------------- /tests/skt/__init__.py: -------------------------------------------------------------------------------- 1 | # coding: utf-8 2 | # 2021/8/25 @ tongshiwei 3 | -------------------------------------------------------------------------------- /tests/skt/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/skt/conftest.py -------------------------------------------------------------------------------- /tests/skt/test_mskt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bigdata-ustc/XKT/HEAD/tests/skt/test_mskt.py --------------------------------------------------------------------------------