Skip to content

Instantly share code, notes, and snippets.

@danwoods
Created December 6, 2019 02:39
Show Gist options
  • Save danwoods/4cbb7370ebe0d8ecb47cbba3fbedc494 to your computer and use it in GitHub Desktop.
Save danwoods/4cbb7370ebe0d8ecb47cbba3fbedc494 to your computer and use it in GitHub Desktop.
import numpy as np
from sklearn.linear_model import LinearRegression
X = np.array([[4, 1, 0], [6, 1, 0], [0, 0, 1]])
y = np.array([90000, 120000, 60000])
model = LinearRegression()
model.fit(X, y)
model.predict(np.array([[5, 1, 0]]))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment