Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP] Wide & Deep migration to PyTorch #2168

Draft
wants to merge 11 commits into
base: staging
Choose a base branch
from
Draft

[WIP] Wide & Deep migration to PyTorch #2168

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
551bf33
[WIP] Started Wide & Deep pytorch migration
daviddavo Sep 18, 2024
756ee6d
Working wide_and_deep pytorch module
daviddavo Sep 20, 2024
fc49b13
Added support for additional embeddings in wide and deep pytorch module
daviddavo Sep 20, 2024
d5e461e
Removed old pytorch model
daviddavo Sep 20, 2024
9bd7d25
Added [start quote] hashed [end quote] cross features
daviddavo Sep 24, 2024
81a8d27
Created WideandDeep wrapper class
daviddavo Sep 29, 2024
5829d16
Added continuous features (genres) to wide and deep
daviddavo Oct 1, 2024
3df2dfe
Save wide and deep model
daviddavo Oct 6, 2024
cc1e2f9
Speedup 10x `WideAndDeep._get_uip_cont`
daviddavo Oct 6, 2024
f713841
WideDeep Avoid eval every iter
daviddavo Oct 6, 2024
2dd40c0
Moved test_wide_deep_utils to pytorch
daviddavo Oct 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading