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.

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

Allow Continuation of Training #148

Open
jtpdowns opened this issue Mar 28, 2023 · 2 comments
Open

Allow Continuation of Training #148

jtpdowns opened this issue Mar 28, 2023 · 2 comments
Labels
new feature Feature request to work on

Comments

@jtpdowns
Copy link

It appears that the fit method for ensembles is also where the estimators are instantiated. It would be convenient (for example for fine-tuning pretrained ensembles) if the instantiation and training happened in separate steps. Would it be possible either to decouple the instantiation and training steps to allow for the continuation of training? Is the functionality for continuation of training already available in some other way?

@jtpdowns
Copy link
Author

It seems like this might be straightforward to implement for any class where all estimators are initialized at once (ie I think adversarial, bagging, fusion, gradient boosting, soft gradient boosting, and voting):

    # Instantiate a pool of base estimators, optimizers, and schedulers.
    estimators = []
    for _ in range(self.n_estimators):
        estimators.append(self._make_estimator())

For fast geometric and snapshot, it seems like you could still manage a list and just update from the last element in the list for continuation (instantiating the first estimator into an otherwise empty list for a new ensemble).

@xuyxu
Copy link
Member

xuyxu commented Mar 29, 2023

Hi @jtpdowns, I think you are right. It would be convenient if we could decouple the training part and the model initialization part. Will appreciate a pull request very much.

@xuyxu xuyxu added the new feature Feature request to work on label Mar 29, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
new feature Feature request to work on
Projects
None yet
Development

No branches or pull requests

2 participants