Skip to main content

Package transforms your categorical variables into embedded vectors. You should have tensorflow, pandas, numpy, keras and sklearn installed.

Attributes: model = EntityEmbedding(dataframe, features from the copy of the df, target column, column you want a vector for)

Hyperparameters you can optimize: model.train_fit(activation1='relu', activation2='relu', activation3='relu', loss='mean_squared_error', metrics='mape', dense_size_num=128, dense_size_conc_1=300, dense_size_conc_2=300, alpha=1e-3, epochs=1000, batch_size=512, verbose=1, patience=5)

Inside model.transform(), always provide embedded vector you want to use: model.transform(model.ent_emb)

model.visualize() returns 2 d visualization of your column categories.

Metadata

Release files for ent-embedding 0.0.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Built distribution (wheel)

Table of built distributions (wheels) for ent-embedding 0.0.1
File Interpreter ABI Platform
ent_embedding-0.0.1-py3-none-any.whl Python 3 none any Details

Release files / ent_embedding-0.0.1-py3-none-any.whl

Download URL ent_embedding-0.0.1-py3-none-any.whl
Size 2.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
b6d2e40113bf1693f6a2239879b51b9ff1fa2b343f65678bb85abbef5105f9cf
BLAKE2b-256 checksum
How to use checksums
c6421e23ee248235473ca8703588c5cb1a1d8d6d45bb13c9f060d9601273ec71
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.8.0 pkginfo/1.8.2 readme-renderer/32.0 requests/2.27.1 requests-toolbelt/0.9.1 urllib3/1.26.8 tqdm/4.62.3 importlib-metadata/4.10.1 keyring/23.5.0 rfc3986/2.0.0 colorama/0.4.4 CPython/3.10.2

Release history Release notifications | RSS feed

This release

0.0.1 This release

1 release file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page