lamp-data
API
Generated with
lamp-data/lamp.data/GraphBatchStream

GraphBatchStream

lamp.data.GraphBatchStream$
object GraphBatchStream

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Self type

Members list

Type members

Classlikes

case class Graph(nodeFeatures: STen, edgeFeatures: STen, edgeI: STen, edgeJ: STen)

Attributes

Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all

Value members

Concrete methods

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

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

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

Returns a pair of node features, edge list

Attributes

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

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

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

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

Attributes

Generated with