SGD_Train.java WIP