added option for neural net regression