I think there is a problem with your feature parameter. if you have two columns in your dataset. Your feature parameter should not be 1D. So check the shape of that and if it is 1D, then convert it from 1D to 2D. Say your feature parameter is x.