Feature Embedding

A sample project to convert feature vectors into embeddings using MovieLens dataset.

Setup

# Download MovieLens dataset
mkdir data
curl https://files.grouplens.org/datasets/movielens/ml-latest-small.zip > data/ml-latest-small.zip
unzip -d data data/ml-latest-small.zip

Installation

pip install .[example]

Run

python examples/train.py

Result

Uninstallation

pip uninstall embedding