本文整理汇总了Java中org.apache.spark.mllib.linalg.Matrix.numCols方法的典型用法代码示例。如果您正苦于以下问题:Java Matrix.numCols方法的具体用法?Java Matrix.numCols怎么用?Java Matrix.numCols使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类org.apache.spark.mllib.linalg.Matrix
的用法示例。
在下文中一共展示了Matrix.numCols方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Java代码示例。
示例1: matrixEquals
import org.apache.spark.mllib.linalg.Matrix; //导入方法依赖的package包/类
private boolean matrixEquals(Matrix mlMatrix, INDArray indMatrix, Double eps) {
final int mlRows = mlMatrix.numRows();
final int mlCols = mlMatrix.numCols();
final int indRows = indMatrix.rows();
final int indCols = indMatrix.columns();
if (mlRows != indRows)
return false;
if (mlCols != indCols)
return false;
for (int i = 0; i < mlRows; i++) {
for (int j = 0; j < mlCols; j++) {
double delta = Math.abs(mlMatrix.apply(i, j) - indMatrix.getDouble(i, j));
if (delta > eps)
return false;
}
}
return true;
}
示例2: convertSparkMatrixToRealMatrix
import org.apache.spark.mllib.linalg.Matrix; //导入方法依赖的package包/类
/**
* Convert a local (not distributed) Spark Matrix to an Apache Commons matrix.
*
* @param r Never {@code null}
* @return Not {@code null}
*/
public static RealMatrix convertSparkMatrixToRealMatrix(final Matrix r){
final RealMatrix result = new Array2DRowRealMatrix(r.numRows(), r.numCols());
final double [] columnMajorMat = r.toArray();
for (int i = 0; i < r.numRows(); i++) {
result.setRow(i, Arrays.copyOfRange(columnMajorMat, i * r.numCols(), i * r.numCols() + r.numCols()) );
}
return result;
}