Allow specification of class col label in extract_training_data function.