Class

net.sansa_stack.ml.spark.kge.linkprediction.models

TransE

Related Doc: package models

Permalink

class TransE extends Models

TransE embedding model ----------------------

Bordes, Antoine, et al. Translating embeddings for modeling multi-relational data. NIPS. 2013.

Created by lpfgarcia on 14/11/2017.

Linear Supertypes
Models, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. TransE
  2. Models
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TransE(train: Dataset[IntegerTriples], ne: Int, nr: Int, batch: Int, k: Int, margin: Float, L: String, sk: SparkSession)

    Permalink

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  4. def L1(vec: Tensor[Float]): Float

    Permalink
    Definition Classes
    Models

  5. def L2(vec: Tensor[Float]): Float

    Permalink
    Definition Classes
    Models

  6. val Ne: Int

    Permalink
    Definition Classes
    Models

  7. val Nr: Int

    Permalink
    Definition Classes
    Models

  8. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any

  9. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  10. def dist(row: IntegerTriples): Tensor[Float]

    Permalink

  11. def dist(data: Dataset[IntegerTriples]): Float

    Permalink

  12. var e: Tensor[Float]

    Permalink
    Definition Classes
    Models

  13. val epochs: Int

    Permalink

  14. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  15. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  16. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  17. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

  18. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  19. def initialize(size: Int): Tensor[Float]

    Permalink
    Definition Classes
    Models

  20. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

  21. val myL: (Tensor[Float]) ⇒ Float

    Permalink

  22. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  23. def negative(data: Dataset[IntegerTriples]): Dataset[IntegerTriples]

    Permalink
    Definition Classes
    Models

  24. def normalize(data: Tensor[Float]): Tensor[Float]

    Permalink
    Definition Classes
    Models

  25. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  26. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  27. var opt: Adam[Float]

    Permalink

  28. var r: Tensor[Float]

    Permalink
    Definition Classes
    Models

  29. val rate: Float

    Permalink

  30. def run(): Unit

    Permalink

  31. val seed: Random

    Permalink
    Definition Classes
    Models

  32. def subset(data: Dataset[IntegerTriples]): Dataset[IntegerTriples]

    Permalink
    Definition Classes
    Models

  33. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef

  34. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any

  35. def tuple(aux: IntegerTriples): IntegerTriples

    Permalink
    Definition Classes
    Models

  36. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  37. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  38. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Models

Inherited from AnyRef

Inherited from Any

Ungrouped