import numpy as np from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) y = np.array([6, 15, 24]) X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0) model = LinearRegression() model.fit(X_train, y_train) y_pred = model.predict(X_test) print(y_pred)