added logging of number of test cases trained