Remove unused code in classifier_test
Remove unsed include