lamp-core
API
Generated with
lamp-core/lamp/lamp.nn/lamp.nn.graph

lamp.nn.graph

package lamp.nn.graph

Members list

Type members

Classlikes

case class GCN[M <: Module](transform: M & Module) extends GraphModule

Attributes

Companion
object
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
object GCN

Attributes

Companion
class
Supertypes
trait Product
trait Mirror
class Object
trait Matchable
class Any
Self type
GCN.type
case class Graph(nodeFeatures: Variable, edgeFeatures: Variable, edgeI: STen, edgeJ: STen, vertexPoolingIndices: STen)

Attributes

Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
case class GraphAttention(wNodeKey1: Constant, wNodeKey2: Constant, wEdgeKey: Constant, wNodeValue: Constant, wAttention: Option[Constant], nonLinearity: Boolean, dropout: Dropout, numHeads: Int) extends GenericModule[Graph, Graph]

Attributes

Companion
object
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
object GraphAttention

Attributes

Companion
class
Supertypes
trait Product
trait Mirror
class Object
trait Matchable
class Any
Self type
case class MPNN[M1 <: Module, M2 <: Module](messageTransform: M1 & Module, vertexTransform: M2 & Module, degreeNormalizeI: Boolean, degreeNormalizeJ: Boolean, aggregateJ: Boolean) extends GraphModule

Attributes

Companion
object
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
object MPNN

Attributes

Companion
class
Supertypes
trait Product
trait Mirror
class Object
trait Matchable
class Any
Self type
MPNN.type
object VertexPooling

Attributes

Supertypes
class Object
trait Matchable
class Any
Self type

Types

Generated with