Skip to content

Commit 756be88

Browse files
committed
-- refined backbone structure
-- made changes to make saving and loading models work -- saving and loading with custom parameters also work -- added MDN tutorial -- added test cases to check custom parameters
1 parent b49f827 commit 756be88

File tree

15 files changed

+1800091
-385498
lines changed

15 files changed

+1800091
-385498
lines changed

docs/tutorials/07-Probabilistic Regression with MDN.ipynb

Lines changed: 796806 additions & 0 deletions
Original file line numberDiff line numberDiff line change

examples/regression_with_MDN.ipynb

Lines changed: 1003063 additions & 385449 deletions
Original file line numberDiff line numberDiff line change

examples/to_test_regression.py

Lines changed: 48 additions & 9 deletions
Original file line numberDiff line numberDiff line change

mkdocs.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change

pytorch_tabular/config/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change

pytorch_tabular/models/autoint/autoint.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change

pytorch_tabular/models/base_model.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change

pytorch_tabular/models/category_embedding/category_embedding_model.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change

pytorch_tabular/models/mixture_density/config.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change

pytorch_tabular/models/mixture_density/mdn.py

Lines changed: 18 additions & 10 deletions
Original file line numberDiff line numberDiff line change

0 commit comments

Comments
 (0)