it is not working for x_train.shape = (n,6) here n = number of samples
it is not working for x_train.shape = (n,6) here n = number of samples