Svm Classification Python