Skip to content

Instantly share code, notes, and snippets.

@sadimanna

sadimanna/read_training_labels_csv.py

Last active March 20, 2020 11:17
Show Gist options
  • Save sadimanna/97cd55a454653211e335a7d2e205fe05 to your computer and use it in GitHub Desktop.
Save sadimanna/97cd55a454653211e335a7d2e205fe05 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
read_training_labels_csv.py
train_data = pd.read_csv('training_labels.csv')
Y = train_data[['label']]
kf = KFold(n_splits = 5)
skf = StratifiedKFold(n_split = 5, random_state = 7, shuffle = True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment