lamp-core/lamp.nn.graph

lamp.nn.graph

package lamp.nn.graph

Type members

Classlikes

case class GCN[M <: Module](transform: M & Module) extends GraphModule
Companion:
object
object GCN
Companion:
class
case class Graph(nodeFeatures: Variable, edgeFeatures: Variable, edgeI: STen, edgeJ: STen, vertexPoolingIndices: STen)
case class GraphAttention(wNodeKey1: Constant, wNodeKey2: Constant, wEdgeKey: Constant, wNodeValue: Constant, wAttention: Option[Constant], nonLinearity: Boolean, dropout: Dropout, numHeads: Int) extends GenericModule[Graph, Graph]
Companion:
object
object GraphAttention
Companion:
class
case class MPNN[M1 <: Module, M2 <: Module](messageTransform: M1 & Module, vertexTransform: M2 & Module, degreeNormalizeI: Boolean, degreeNormalizeJ: Boolean, aggregateJ: Boolean) extends GraphModule
Companion:
object
object MPNN
Companion:
class
object VertexPooling

Types