MAHOUT-1738: implement MatrixFlavor for H2OBlockMatrix Closes apache/mahout#140
Signed-off-by: Anand Avati <[email protected]> Project: http://git-wip-us.apache.org/repos/asf/mahout/repo Commit: http://git-wip-us.apache.org/repos/asf/mahout/commit/273ce56e Tree: http://git-wip-us.apache.org/repos/asf/mahout/tree/273ce56e Diff: http://git-wip-us.apache.org/repos/asf/mahout/diff/273ce56e Branch: refs/heads/mahout-0.10.x Commit: 273ce56efa8a1180f22206844038daf2f2c4cebf Parents: 8a6b805 Author: Anand Avati <[email protected]> Authored: Mon Jun 22 15:39:21 2015 -0700 Committer: Anand Avati <[email protected]> Committed: Mon Jun 22 20:01:19 2015 -0700 ---------------------------------------------------------------------- .../org/apache/mahout/h2obindings/H2OBlockMatrix.java | 12 ++++++++++++ 1 file changed, 12 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/mahout/blob/273ce56e/h2o/src/main/java/org/apache/mahout/h2obindings/H2OBlockMatrix.java ---------------------------------------------------------------------- diff --git a/h2o/src/main/java/org/apache/mahout/h2obindings/H2OBlockMatrix.java b/h2o/src/main/java/org/apache/mahout/h2obindings/H2OBlockMatrix.java index a073e73..378f7b6 100644 --- a/h2o/src/main/java/org/apache/mahout/h2obindings/H2OBlockMatrix.java +++ b/h2o/src/main/java/org/apache/mahout/h2obindings/H2OBlockMatrix.java @@ -22,6 +22,7 @@ import org.apache.mahout.math.Vector; import org.apache.mahout.math.AbstractMatrix; import org.apache.mahout.math.DenseMatrix; import org.apache.mahout.math.SparseMatrix; +import org.apache.mahout.math.flavor.MatrixFlavor; import water.fvec.Chunk; @@ -119,4 +120,15 @@ public class H2OBlockMatrix extends AbstractMatrix { cow.assignColumn(col, v); return cow; } + + @Override + public MatrixFlavor getFlavor() { + if (cow != null) { + return cow.getFlavor(); + } else if (chks[0].isSparse()) { + return MatrixFlavor.SPARSELIKE; + } else { + return MatrixFlavor.DENSELIKE; + } + } }
