/keras_entity_embedding

Keras NN model with entity embedding for categorical inputs

Primary LanguagePythonMIT LicenseMIT

Keras Entity Embedding

Utility function to create a NN model in Keras with entity embedding for the categorical inputs.

Logic and heuristic parameters derived from fast.ai.