I\'ve successfully used this for classification, but I want to apply it to a regression problem. I understand that I should be using MSE instead of entropy, but I don\'t see