Skip to content

Instantly share code, notes, and snippets.

@djsegal
Created December 30, 2019 03:00
Show Gist options
  • Save djsegal/8ef31dfc8e3e861c363e858b359f71d9 to your computer and use it in GitHub Desktop.
Save djsegal/8ef31dfc8e3e861c363e858b359f71d9 to your computer and use it in GitHub Desktop.
import numpy as np
from sklearn.compose import TransformedTargetRegressor
class LogTransformedTargetRegressor(TransformedTargetRegressor):
def __init__(self, regressor):
inverse_func = lambda cur_log_val: 10 ** cur_log_val
TransformedTargetRegressor.__init__(
self, regressor=regressor,
func=np.log10, inverse_func=inverse_func
)
@property
def feature_importances_(self):
return self.regressor_.feature_importances_
@property
def coef_(self):
return self.regressor_.coef_
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment