Will use F1_score for evaluation