Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created June 21, 2021 12:57
Show Gist options
  • Save amankharwal/1dc0567071f10faaecf937369ad7254f to your computer and use it in GitHub Desktop.
Save amankharwal/1dc0567071f10faaecf937369ad7254f to your computer and use it in GitHub Desktop.
x = data[["Open", "High", "Low"]]
y = data["Close"]
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