Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created April 29, 2021 08:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/21497d4350f3a827e9f3ec34486672b3 to your computer and use it in GitHub Desktop.
Save amankharwal/21497d4350f3a827e9f3ec34486672b3 to your computer and use it in GitHub Desktop.
from sklearn.linear_model import LinearRegression
model = LinearRegression()
model.fit(xtrain, ytrain)
ypred = model.predict(xtest)
data = pd.DataFrame(data={"Predicted Profit": ypred.flatten()})
print(data.head())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment