moved test_classifier to optimizations.py
added get_prediction functionality to optimizations
removed y_pred from list