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

Jump to bottom

Support model saving and loading #316

Merged
merged 73 commits into from
Feb 6, 2020
Merged

Support model saving and loading #316

merged 73 commits into from
Feb 6, 2020

Conversation

tqtg
Copy link
Member

@tqtg tqtg commented Feb 4, 2020 •
edited
Loading

Description

  • Support model saving and loading via save and load APIs
  • Support to save the whole Experiment via save_dir argument
  • All examples are tested on Linux (expected to work on Windows and Mac OS as well)

Related Issues

Checklist:

  • I have added tests.
  • I have updated the documentation accordingly.
  • I have updated README.md (if you are adding a new model).
  • I have updated examples/README.md (if you are adding a new example).
  • I have updated datasets/README.md (if you are adding a new dataset).

@tqtg
Support simple model saving and loading using pickle
e046da8
@tqtg
Add save_dir argument to the Experiment
352ae62
@tqtg
Tested BPR and WBPR models
4911342
@tqtg
Tested MF model
14f8bd2
@tqtg
Tested BaselineOnly model
5b0142f
@tqtg
Refactor code
5ce79aa
@tqtg
Remove access to train_set and val_set when saving a model
00cad4a
@tqtg
Tested CDL model
045c773
@tqtg
Tested CDR model
c3f95bb
@tqtg
Refactor code
95c081a
@tqtg
Refactor COE
015b56c
@tqtg
Tested ConvMF model
5cd4373
@tqtg
Tested CTR model
3e4a038
@tqtg
Refactor code
b75814d
@tqtg
Tested CVAE model
3228ef3
@tqtg
Tested EFM model
05d3116
@tqtg
Refactor code
62df655
@tqtg
Refactor code
c319232
@tqtg
Tested HFT model
a0234b9
@tqtg
Tested HPF model
ee3fa32
@tqtg
Refactor IBPR
d44f270
@tqtg
Tested MCF model
1f8283c
@tqtg
Tested MostPop
08d15f7
@tqtg
Tested MTER model
ab334a3
@tqtg
Refactor and test NCF models
3d3b7c7
@tqtg
Refactor NCF
b9ab26b
@tqtg
Refactor NCF
8eee7c1
@tqtg
Update model save format
e284b6a
@tqtg
Update saving format
8d7521f
@tqtg
Refactor C2PF model
0ab216a
@tqtg
Tested VBPR model
919abcb
@tqtg
Tested VAECF model
e1b8f1d
@tqtg
Add init_params as a default attribute
7a890a4
@tqtg
Fix init_params is None for some models
c3ed515
@tqtg
Instantiate a completely new model for each fold of CrossValidation
be8585a
@tqtg
Fix format str error on Windows
7fc420c
@tqtg
Fix format str error on Windows
01a4f49
@tqtg
Rename lr attribute in NCF models
1173be5
@tqtg
Refine clone function of Recommender
94cf857
@tqtg
Disable model saving when running Experiment with CrossValidation
5e49ab5
@tqtg
Standardize init_params in C2PF model
68b6f76
@tqtg
Refactor CDL
46d3ab2
@tqtg
Refactor CDR
4354b11
@tqtg
Standardize init_params in COE model
90f4dd9
@tqtg
Standardize init_params in HPF model
6b7c556
@tqtg
Standardize init_params in COE model
5da1821
@tqtg
Standardize init_params in IBPR model
0a76069
@tqtg
Standardize init_params in MCF model
f5898eb
@tqtg
Standardize init_params in IBPR model
aa63844
@tqtg
Standardize init_params in PCRL model
b1d3a9f
@tqtg
Standardize init_params in PMF model
1f8b8ac
@tqtg
new_params is optional when cloning a Recommender
fd616b9
@tqtg
Refactor skmeans
4bea84a
@tqtg
Standardize init_params in SoRec model
d41214a
@tqtg
Standardize init_params in VMF model
c8cd715
@tqtg
Standardize init_params in VBPR model
d410b06
@tqtg
Add requires_grad for torch Tensor from_numpy
66cdfb7
@tqtg tqtg merged commit d203d84 into PreferredAI:master Feb 6, 2020
@tqtg tqtg deleted the save-load-model branch February 6, 2020 03:12
tqtg added a commit to amirj/cornac that referenced this pull request May 22, 2021
@tqtg
Support model saving and loading (PreferredAI#316)
1cd5f74
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@tqtg