added function for test/train sets
added function for test/train sets

Neuroglycerin/hail-seizure,Neuroglycerin/hail-seizure,Neuroglycerin/hail-seizure,Neuroglycerin/hail-seizure,Neuroglycerin/hail-seizure,Neuroglycerin/hail-seizure