Make sure to divide data before into traing set and testing set.
plt.scatter(train.INDEPEDANTVARIABLENAME, train.DEPENDANTVARIABLENAME, color='COLOROFDATAPOINTS') plt.xlabel("INDEPENDANTCHARTLABEL") plt.ylabel("DEPENDANTCHARTLABEL") plt.show()