Add a default case-base and fix method param renaming bug in TrainAction

Signed-off-by: MHesham <mhesham.fcis@gmail.com>