We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Artificial Intelligence
  3. Statistics and Machine Learning
  4. Day 6: Multiple Linear Regression: Predicting House Prices
  5. Discussions

Day 6: Multiple Linear Regression: Predicting House Prices

  • bhimandey
     + 0 comments

    Multiple Linear Regression: Predicting House Prices Incase you need some help.

    import numpy as np
from sklearn.linear_model import LinearRegression

F, N = map(int, input().split())
data = [list(map(float, input().split())) for i in range(N)]
data = np.array(data)

X_train, y_train = data[:, :-1], data[:, -1]

T = int(input())

X_test = [list(map(float, input().split())) for i in range(T)]
X_test = np.array(X_test)

model = LinearRegression()
model.fit(X_train, y_train)

predictions = model.predict(X_test)

for result in predictions:
    print(round(result, 2))