1poly = PolynomialFeatures(degree=2)
2X_F1_poly = poly.fit_transform(X_F1)
3
4X_train, X_test, y_train, y_test = train_test_split(X_F1_poly, y_F1,
5 random_state = 0)
6linreg = LinearRegression().fit(X_train, y_train)
7