Packages

t

ai.eto.rikai.sql.spark

SparkRunnable

trait SparkRunnable extends AnyRef

Make ai.eto.rikai.sql.model.Model runnable on Spark.

For a ML_PREDICT expression in Spark SQL,

SELECT ML_PREDICT(model_zoo, col1, col2, col3) FROM t1

It generates a LogicalPlan equivalent to

SELECT <Model{model_zoo}.asSpark(col1, col2, col3)> FROM t1
Example:

  1. To implement a ai.eto.rikai.sql.model.Model for RegistryFoo:

    class FooModel(name, uri) extends Model with SparkRunnable {

    /** Use a Spark UDF with the same name to run RegistryFoo's model */
   def asSpark(args: Seq[Expression]) : Expression = {
       UnresolvedFunction(
          new FunctionIdentifier(s"${name}"),
          arguments,
          isDistinct = false,
          Option.empty
       )
   }
Note

It is the ai.eto.rikai.sql.model.Registry's responsibility to implement a Model that runs with Spark SQL.

Linear Supertypes
AnyRef, Any
Known Subclasses
SparkUDFModel
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. SparkRunnable
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def asSpark(args: Seq[Expression]): Expression

    Convert a ai.eto.rikai.sql.model.Model to a Spark Expression in Spark SQL's logical plan.

Concrete Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  6. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  10. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  11. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  12. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  13. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  14. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  15. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  16. def toString(): String
    Definition Classes
    AnyRef → Any
  17. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  18. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  19. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from AnyRef

Inherited from Any

Ungrouped