Skip to content

Instantly share code, notes, and snippets.

Created Apr 29, 2021
What would you like to do?
x = data[["R&D Spend", "Administration", "Marketing Spend"]]
y = data["Profit"]
x = x.to_numpy()
y = y.to_numpy()
y = y.reshape(-1, 1)
from sklearn.model_selection import train_test_split
xtrain, xtest, ytrain, ytest = train_test_split(x, y, test_size=0.2, random_state=42)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment