class Spark::Mllib::DenseMatrix

DenseMatrix

DenseMatrix.new(2, 3, [[1,2,3], [4,5,6]]).values
# => [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]

Public Class Methods

from_java(object) click to toggle source
# File lib/spark/mllib/matrix.rb, line 51
def self.from_java(object)
  rows = object.numRows
  cols = object.numCols
  values = object.values

  DenseMatrix.new(rows, cols, values)
end
new(rows, cols, values) click to toggle source
Calls superclass method
# File lib/spark/mllib/matrix.rb, line 43
def initialize(rows, cols, values)
  super(:dense, rows, cols, values.to_a)
end

Public Instance Methods

to_java() click to toggle source
# File lib/spark/mllib/matrix.rb, line 47
def to_java
  JDenseMatrix.new(shape[0], shape[1], values.flatten)
end