search

saattrupdan / doubt

Bringing back uncertainty to machine learning.
MIT License
50 stars 3 forks source link

Attribute feature_importances_ #32

Closed rrealrangel closed 2 years ago

rrealrangel commented 2 years ago

Hi. First of all, I hope this is the correct way to point this out. As a user, I am not familiar with developer protocols. I am using the QuantileRegressionForest class and have noticed that the 'featureimportances' attribute is not available for my trained model. Is there any way to access this attribute? Do you plan to incorporate it in a future version? Thanks in advance. (By the way, thanks for sharing the module).

saattrupdan commented 2 years ago

Hi, and thanks for your interest in doubt!

I agree that this would be a good feature to have. The individual trees should support this already, but the forests do not. We could use the original sklearn implementation here. I'll implement it as soon as I got the time 🙂

saattrupdan commented 2 years ago

@rrealrangel Implemented, after quite a long delay, in v4.3.0 🎉