Skip to content

Latest commit

 

History

History
18 lines (13 loc) · 657 Bytes

README.md

File metadata and controls

18 lines (13 loc) · 657 Bytes

Entity Embedding for Gaussian Process Regression

This code provides a method to represent categorical data by learned embedding vectors in the Gaussian Process regression of scikit-learn. Each level of the categorical variable is seen as an entity that is associated with a vector. The embedding vectors are learned from data.

Installation

Inside this folder run pip install .

Example and Usage

The file examples/MinimalExample.ipynb contains a minimal example.
The notebook requires some more dependencies:

In the folder examples run pip install -r requirements.txt.

Then run jupyter lab