这是indexloc提供的服务,不要输入任何密码
Skip to content
This repository was archived by the owner on Aug 3, 2021. It is now read-only.
This repository was archived by the owner on Aug 3, 2021. It is now read-only.

Add MovieLens example #2

@okuchaiev

Description

@okuchaiev

Provide a recepie for training a model on MovieLens data (20M and 1M).

This should include the following:

  1. 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.
  1. An example train.sh, and test.sh for 20M and 1M MovieLens data sets which will train/eval and test the model correspondingly.

  2. Choose model architecture and hyper parameters that give best eval RMSE you can get.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions