Changed mat64.Matrix to train.RowMatrix