Need Help at the Bob the Builder Task

Question

I have tried this solution. 4 out of 6 test cases are working but the 3rd and 6th are not working. Can anyone please help me with a new solution?

n = int(input())
X = []
for i in range(n):
X.append([float(x) for x in input().split()])
y = [int(x) for x in input().split()]
datapoint = [float(x) for x in input().split()]

import numpy as np
from sklearn.linear_model import LinearRegression

lreg = LinearRegression().fit(X, y)
y_pred = lreg.predict([datapoint])

print(int(y_pred[0]))

Answer 1

use LogisticRegression. It is stated in the problem.

Answer 2

Well… yeah thank you.

Answer 3

from sklearn.linear_midel
SyntaxError : invalided syntax
Can someone explain this why?