enhancementgood first issuehelp wanted
倉庫指標
- 星標
- (1,702 顆星)
- PR 合併指標
- (PR 指標待抓取)
描述
Provide a recepie for training a model on MovieLens data (20M and 1M).
This should include the following:
- Data converter. Consider fixing/adjusting this script
- Train/Eval/Test split. Training test ratings should come before any Eval and Test ratings. See section 3.1 (Experiment setup) of this paper for details and Netflix converter for example.
-
An example train.sh, and test.sh for 20M and 1M MovieLens data sets which will train/eval and test the model correspondingly.
-
Choose model architecture and hyper parameters that give best eval RMSE you can get.