Adding interface and function for gradiant boosting