java - RowMatrix、MLlib、Java Spark

Question

我有RowMatrix我的问题是，我如何通过索引来操纵它？这个问题与这个问题非常相似:

Matrix Operation in Spark MLlib in Java

最后，我需要的就是拥有带有良好类库的 Matrix。目前我无法操作这个对象。

Answer 1

正如RowMatrix的JavaDocs所示

:: Experimental :: Represents a row-oriented distributed Matrix with no meaningful row indices.

行上没有顺序。您可以通过调用 toBreeze 从中获取 breeze.linalg.DenseMatrix，但不能保证行的顺序。当它们到达主设备时，它们只是被插入到结果矩阵中。这意味着此操作的结果可能会不时发生变化。

如果您需要 toBreeze 操作的确定性结果，则必须使用 IndexedRowMatrix。每行都分配了一个行索引，用于构建breeze.linalg.DenseMatrix。

从那里您可以使用建议的解决方案 here ，即

import no.uib.cipr.matrix.DenseMatrix;
// ...
IndexedRowMatrix U = svd.U();
DenseMatrix U_mtj = new DenseMatrix((int) U.numCols(), (int) U.numRows(), U.toBreeze().toArray$mcD$sp(), true);
// From there, matrix operations are available on U_mtj