embedding-based item nearest neighborhoods extraction
Project description
DeepNeighbor
DeepNeighbor is a High-level,Flexible and Extendible package for embedding-based information retrieval from user-item interaction logs. Just as the name suggested, 'deep' means deep learning models to get user/item embeddings, while 'neighbor' means approximate nearest neighbor search in the embedding space.
It mainly has two parts : Embed step and Search step by the following codes:
model.train()
,which generates embeddings for users and items (Deep),
model.search()
, which looks for Approximate nearest neighbor for seed user/item (Neighbor) .
Install
pip install deepneighbor
How To Use
from deepneighbor.embed import Embed
model = Embed(data)
model.train()
model.search(seed = 'Louis', k=10)
Input format
The input data for the Embed() should be a pandas DataFrame with two columns in order: 'user' and 'item'. For each user, the item are recommended to be ordered by time.
Models
- word2vec
- Siamese Network with triple loss
- deepwalk
- graph convolutional network
- matrix factorization
- graph attention network
Examples
TBD
License
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for deepneighbor-0.2.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 247e5e8ed2efb2932d4e193f0c0fc01133c8d9d561b775a629defffecdbbf1d4 |
|
MD5 | 7b46cda025d0eec914def7465aa5da7a |
|
BLAKE2b-256 | b3aaed3ee455325b025e2ee8b969309a91c7960e9cf56cd80e6013d028aaf2e2 |