Clean up linear classifier
Use mini-batch SGD to train linear classifier