Professional Documents
Culture Documents
1_Linear Regression
1_Linear Regression
1_Linear Regression
import pandas as pd
data = pd.read_csv('C:/Users/ARYA/Salary_Data.csv')
print(data.head())
print(data.shape)
print(data.isnull().sum())
x = data.iloc[:, :1].values
y = data.iloc[:, 1:2].values
print(x)
print(y)
model = LinearRegression()
model.fit(x_train, y_train)
y_pred = model.predict(x_test)
print(y_pred)
print(y_test)
plt.xlabel('Experience in Years')
plt.ylabel('Salary in Rupees')
plt.show()
plt.xlabel('Experience in Years')
plt.ylabel('Salary in Rupees')
plt.show()