Darel13712 / ease_rec

embarrassingly shallow autoencoder

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

EASE recommender

PWC PWC PWC

This is a repository for a model from Embarrassingly Shallow Autoencoders for Sparse Data.

This model is cool because it is a closed-form solution.

Similiarity matrix is calculated on CPU with numpy.

Input

pandas.DataFrame with columns user_id and item_id both for fit and predict.

It may also use ratings from column rating if implicit parameter is set to False.

Output

pandas.DataFrame with columns user_id, item_id, score

About

embarrassingly shallow autoencoder


Languages

Language:Python 100.0%