Description:
product2vec 0.1.0
product2vec
product2vec is a Python library that implements Product2Vec model. It is capable of finding complements and substitutes among products given shopping baskets. Current implementation is based on the original paper: https://ssrn.com/abstract=3519358
Installation
pip install product2vec
Usage
>>> from product2vec import Product2Vec
>>> # toy dataset with two baskets and 4 products
>>> data = [
... ['coffee', 'cookies', 'chocolate'],
... ['tea', 'cookies', 'chocolate'],
... ]
>>> prod2vec = Product2Vec(vector_size=3, min_count=1, sample=0, seed=1, workers=1)
>>> _ = prod2vec.fit(data)
>>> prod2vec.show_substitutes(product='tea', topn=2)
[('coffee', 0.024425969), ('chocolate', 0.023691988)]
>>> prod2vec.show_complements(product='cookies', topn=2)
[('chocolate', 0.5030633), ('coffee', 0.5007087)]
Refer to usage_example.ipynb which can be found in GitHub repository for short model description and elaborate usage.
Contributing
If you spot any bugs or have suggestions don't hesitate to open an issue.
License
For personal and professional use. You cannot resell or redistribute these repositories in their original state.
Customer Reviews
There are no reviews.
