@Override public FloatMatrix getInitTheta() { FloatMatrix x = FloatMatrix.rand(rows, features); FloatMatrix theta = FloatMatrix.rand(columns, features); return MatrixUtil.merge(x.data, theta.data); }