I am getting the below value due to dimension issue. My input features are ColA and ColB and I want to predict the ColC. When i try to fit the model i am getting the below issue