Kernel Methods Sklearn Train Test Split Stratify