Use matrix abstraction for LogisticRegression coefficients during training

This is a follow up from SPARK-18060. The current code for logistic regression relies on manually indexing flat arrays of column major coefficients, which can be messy and is hard to maintain. We can use a matrix abstraction instead of a flat array to simplify things. This will make the code easier to read and maintain.