Added tests for neural net and corrected bug in non-cached predict