import numpy as np import pandas as pd import matplotlib.pyplot as pt data1 = pd.read_csv(\'stage1_labels.csv\') X = data1.iloc[:, :-1].values y = data1.iloc[:, 1]
Print X_train shape. What do you see? I'd bet X_train is 2d (matrix with a single column), while y_train 1d (vector). In turn you get different sizes.
X_train
y_train
I think using X_train[:,0] for plotting (which is from where the error originates) should solve the problem
X_train[:,0]