Item2Vec Wrapped With Python

Gensim 4.x and sklearn

  1. Introduction
  2. General informations
  3. Methods Description
  4. GridSearch and BayesSearch Usage
  5. BayesSearch Example

Prod2Vec or Item2Vec produces embedding for items in a latent space. The method is capable of inferring item-item relations even when user information is not available. It’s based on NLP model Word2Vec. Click here to know more

This project provide a class that encapsulates Item2Vec model (word2vec gensim model) as a sklearn estimator.

It allows the simple and efficient use of the Item2Vec model by providing :

!! Warning : Estimators template is not respected since X does not have the shape (n_features, n_examples)

2.1 Input data format