Put split and extraction function in separate file
Updated to use npy stored trained_classifier and imported functions
Updated notebook