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

Signed-off-by: MHesham <mhesham.fcis@gmail.com>
Merge pull request #110 from MHesham/visializer

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