I wrote a function to train model, but got below error for executing the train function, could not figure out why?
\'<\' not supported between instances of \'Expon