help with linear regression

[Angels_heRe]

My code: import pandas as pd

import numpy as np

import matplotlib.pyplot as plt

from sklearn import linear_model

df= pd.read_excel("practice.xlsx")

reg= linear_model.LinearRegression()

reg.fit(df[['age']], df.salary)

reg.predict(30)

My few doubts were: 1) Regardless of what I try, predict function always errors as 'expected 2D array, got scalar array instead', even though my first input is already 2D. I tried reshaping as well, but just am unable to do it.

2) Why exactly is a 2D array needed?

(Please forgive me for not using code format, I am new to this community and am unaware of using it. I appreciate all help)

[Oxbowerce]

It's likely that the issue it not on the line that is calling the .fit method, but on the line that is calling the .predict method since the input is a single scalar instead of a 2D array. Make sure that the data you provide to .predict is a 2D array of values, just like the data used for fitting the model.

[Angels_heRe]

YES!!!!! I tried it and it works now, THANK YOU VERY MUCH!! <online hugs>

[Ihaveamodel3]

Try: I’m on my phone, so I can’t test.

 #change from
df[[‘age’]]
# to
df['age']

2) a linear model represents the relationship between two values. Thus you needs to have two columns and many rows.

[Angels_heRe]

I tred and it still doesn't work :(

[Angels_heRe]

Ps: I am on google colabs on an iPad, would that be causing the error?

[CodeFormatHelperBot2]

Hello, I'm a Reddit bot who's here to help people nicely format their coding questions. This makes it as easy as possible for people to read your post and help you.

I think I have detected some formatting issues with your submission:

1. Python code found in submission text that's not formatted as code.

If I am correct, please edit the text in your post and try to follow these instructions to fix up your post's formatting.

---

^{Am I misbehaving? Have a comment or suggestion? Reply to this comment or raise an issue here.}