Update next-basket evaluation (#559)
* Fix eval and add batch basket items iteration
* refactor code
* Fix scoring function
* refactor code
* Add unittest for NextBasketEvaluation
* Add unittest for BasketDataset
* Add test NextBasketRecommender
* Add test case reading basket data
* refactor code
* Add history basket ids for accessing extra data in scoring function
* Add option to scoring function based on quantity provided in extra_data
* refactor code
* reuse user_iter() in ub_iter()
* reuse ub_iter() in ubi_iter()
* consider using num_user_batches() instead of overwriting existing num_batches()
---------
Co-authored-by:
tqtg <tuantq.vnu@gmail.com>
Showing
- cornac/data/dataset.py 43 additions, 27 deletionscornac/data/dataset.py
- cornac/eval_methods/next_basket_evaluation.py 28 additions, 39 deletionscornac/eval_methods/next_basket_evaluation.py
- cornac/models/gp_top/recom_gp_top.py 36 additions, 10 deletionscornac/models/gp_top/recom_gp_top.py
- examples/gp_top_tafeng.py 2 additions, 1 deletionexamples/gp_top_tafeng.py
- tests/basket.txt 50 additions, 0 deletionstests/basket.txt
- tests/cornac/data/test_dataset.py 31 additions, 4 deletionstests/cornac/data/test_dataset.py
- tests/cornac/data/test_reader.py 35 additions, 24 deletionstests/cornac/data/test_reader.py
- tests/cornac/eval_methods/test_next_basket_evaluation.py 57 additions, 0 deletionstests/cornac/eval_methods/test_next_basket_evaluation.py
- tests/cornac/models/test_recommender.py 20 additions, 2 deletionstests/cornac/models/test_recommender.py
Loading
Please register or sign in to comment