Train Test Split | Training and Testing data | Machine Learning(ML)

 import pandas as pd

from sklearn.model_selection import train_test_split

from sklearn.linear_model import LinearRegression

data=pd.read_csv('sum.csv')

X=data[['X','Y']]

Y=data['SUM']

X_train,X_test,Y_train,Y_test=train_test_split(X,Y,test_size=.2)

# print(X_train)

# print(Y_train)

model=LinearRegression()

model.fit(X_train,Y_train)

y_pred=model.predict(X_test)

# print(y_pred)

df=pd.DataFrame({'actual':Y_test,'predict':y_pred})

print(df)