org.apache.mahout.sparkbindings.drm.CheckpointedDrmSparkOps.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mahout-spark_2.10 Show documentation

Mahout Bindings for Apache Spark

There is a newer version: 0.13.0

package org.apache.mahout.sparkbindings.drm

import org.apache.mahout.math.drm.CheckpointedDrm
import org.apache.mahout.sparkbindings.DrmRdd

/** Additional Spark-specific operations. Requires underlying DRM to be running on Spark backend. */
class CheckpointedDrmSparkOps[K](drm: CheckpointedDrm[K]) {

  assert(drm.isInstanceOf[CheckpointedDrmSpark[K]], "must be a Spark-backed matrix")

  private[sparkbindings] val sparkDrm = drm.asInstanceOf[CheckpointedDrmSpark[K]]

  /** Spark matrix customization exposure */
  def rdd = sparkDrm.rddInput.asRowWise()

}