I am trying to plot a binary classification using train function. Not matter what method I tried, always return with this error. Please help me out with it.