Skip to content

Latest commit

 

History

History
22 lines (15 loc) · 1.07 KB

indexedrowmatrix.md

File metadata and controls

22 lines (15 loc) · 1.07 KB

IndexedRowMatrix

IndexedRowMatrix與RowMatrix是相似的,但其行十月戈火引具有特定定含義,本質上是一個人戈弓戈口有索引訊息的行數據集合(an RDD of indexed rows)。 每一行由long類型索引和一個本地向量組成。

IndexedRowMatrix可以由一個RDD[IndexedRow]實例被創建,其中IndexedRow是一個被包裝過的(Long, Vcetor)。IndexedRowMatrix可以透過刪除行索引被轉換成RowMatrix。

import org.apache.spark.mllib.linalg.distributed.{IndexedRow, IndexedRowMatrix, RowMatrix}

val rows: RDD[IndexedRow] = ... // an RDD of indexed rows
// Create an IndexedRowMatrix from an RDD[IndexedRow].
val mat: IndexedRowMatrix = new IndexedRowMatrix(rows)

// Get its size.
val m = mat.numRows()
val n = mat.numCols()

// Drop its row indices.
val rowMat: RowMatrix = mat.toRowMatrix()