Add Temporal-Item-Frequency-based User-KNN (TIFUKNN) model (#566)
* Add Temporal-Item-Frequency-based User-KNN (TIFUKNN) model for next basket recommendation * replace KNN with KDTree * refactor code * Fix scoring function using the average neighbors vectors
Showing
- README.md 1 addition, 0 deletionsREADME.md
- cornac/models/__init__.py 1 addition, 0 deletionscornac/models/__init__.py
- cornac/models/tifuknn/__init__.py 16 additions, 0 deletionscornac/models/tifuknn/__init__.py
- cornac/models/tifuknn/recom_tifuknn.py 171 additions, 0 deletionscornac/models/tifuknn/recom_tifuknn.py
- examples/README.md 2 additions, 0 deletionsexamples/README.md
- examples/tifuknn_tafeng.py 45 additions, 0 deletionsexamples/tifuknn_tafeng.py
Loading
Please register or sign in to comment