Changing script so it is possible to specify level1 training set
and level2 training set.

Signed-off-by: Renato Stoffalette Joao <renatosjoao@gmail.com>