adding support for regression model load