thundergbm icon indicating copy to clipboard operation
thundergbm copied to clipboard

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

Open DonSeger opened this issue 4 years ago • 1 comments

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?

DonSeger avatar Jan 23 '22 15:01 DonSeger

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.

Kurt-Liuhf avatar Feb 10 '22 02:02 Kurt-Liuhf