< Back

Packages

package root
Definition Classes
root
package lamp
Definition Classes
root
package nn
Provides building blocks for neural networks
Provides building blocks for neural networks
Notable types:
- nn.GenericModule is an abstraction on parametric functions
- nn.Optimizer is an abstraction of gradient based optimizers
- nn.LossFunction is an abstraction of loss functions, see the companion object for the implemented losses
- nn.SupervisedModel combines a module with a loss
Optimizers:
Modules facilitating composing other modules:
- nn.Sequential composes a homogenous list of modules (analogous to List)
- nn.sequence composes a heterogeneous list of modules (analogous to tuples)
- nn.EitherModule composes two modules in a scala.Either
Examples of neural network building blocks, layers etc:
- nn.Linear implements W X + b with parameters W and b and input X
- nn.BatchNorm, nn.LayerNorm implement batch and layer normalization
- nn.MLP is a factory of a multilayer perceptron architecture
Definition Classes
lamp
package bert
Definition Classes
nn
package graph
Definition Classes
nn
GCN
Graph
GraphAttention
MPNN
VertexPooling

p

lamp.nn

graph

package graph

Linear Supertypes

AnyRef, Any

Ordering

Alphabetic
By Inheritance

Inherited

graph
AnyRef
Any

Hide All
Show All

Visibility

Public
Protected

Type Members

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

Value Members

object GCN extends Serializable
object GraphAttention extends Serializable
object MPNN extends Serializable
object VertexPooling

Inherited from AnyRef

Inherited from Any

Ungrouped