Learning Cognitive Models using Neural Networks