Skip to content

Instantly share code, notes, and snippets.

@PratyushTripathy

PratyushTripathy/splitData.py

Last active September 3, 2019 11:17
Show Gist options
  • Save PratyushTripathy/ece33344ded519a8d6460637b575219a to your computer and use it in GitHub Desktop.
Save PratyushTripathy/ece33344ded519a8d6460637b575219a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
splitData.py
from sklearn.model_selection import train_test_split
xTrain, xTest, yTrain, yTest = train_test_split(featuresBangalore, labelBangalore, test_size=0.4, random_state=42)
print(xTrain.shape)
print(yTrain.shape)
print(xTest.shape)
print(yTest.shape)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment