I ran into the same problem. I fixed it by removing nans in the training data and scaling features. Both StandardScaler() and MinMaxScaler() from sklearn worked cf. https://www.analyticsvidhya.com/blog/2020/04/feature-scaling-machine-learning-normalization-standardization/ 42)