Skip to content

Latest commit

 

History

History
28 lines (23 loc) · 497 Bytes

README.md

File metadata and controls

28 lines (23 loc) · 497 Bytes

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

plot

Uninstallation

pip uninstall embedding