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

AttributeError: 'TGBMClassifier' object has no attribute 'save' #74

Open
DonSeger opened this issue Jan 23, 2022 · 1 comment
Open

AttributeError: 'TGBMClassifier' object has no attribute 'save' #74

DonSeger opened this issue Jan 23, 2022 · 1 comment

Comments

@DonSeger
Copy link

Are there functions in thundergbm that can save a trained model to file and reload it from the file later?

There is a function "clf.predict_proba()" in sklearn that calculates probabilities of the different labels. Is there a corresponding function in thundergbm?

There is a function in sklearn that calculates the importance of the different features, called "clf.feature_importances_". Is there is a similar function in thundergbm?

@Kurt-Liuhf
Copy link
Collaborator

Kurt-Liuhf commented Feb 10, 2022

Hi @DonSeger, You can use clf.save_model(path) and clf.reload_model(path) to save and reload trained models.
But the calculation of probabilities for different labels and the feature importance evaluation are unsupported features in ThunderGBM. We are considering supporting these features in the future, thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants