Fix driver::classifier:train interface