fixing mllib implementation of gradient descent
setting initial weights