Packages

o

lamp.data

GraphBatchStream

object GraphBatchStream

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. GraphBatchStream
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Type Members

  1. case class Graph(nodeFeatures: STen, edgeFeatures: STen, edgeI: STen, edgeJ: STen) extends Product with Serializable

Value Members

  1. def singleLargeGraph(graph: Graph, targetPerNode: STen): BatchStream[(nn.graph.Graph, STen), Boolean, Unit]

    Forms full batches of one big graph.

    Forms full batches of one big graph. Target is on node level

    Returns a pair of node features, edge list

  2. def smallGraphStream(minibatchSize: Int, graphNodesAndEdges: Array[Graph], targetPerGraph: STen, rng: Option[Random], transferBufferSize: Long): BatchStream[(nn.graph.Graph, STen), Int, (BufferPair, BufferPair)]

    Forms minibatches from multiple small graphs.

    Forms minibatches from multiple small graphs. Assumes target is graph level.

    Returns a triple of node features, edge list, graph indices