fix the code, now it can train SVM with different #cells