Merge pull request #24 from CarloLucibello/cl/data2

show for GNNGraph

Author: CarloLucibello

Project.toml

@@ -15,8 +15,6 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"
-PrettyPrint = "8162dcfd-2161-5ef2-ae6c-7681170c5f98"
-PrettyPrinting = "54e16d92-306c-5ea0-a30b-337be88ac337"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"

docs/make.jl

@@ -4,7 +4,7 @@ using Documenter
 DocMeta.setdocmeta!(GraphNeuralNetworks, :DocTestSetup, :(using GraphNeuralNetworks); recursive=true)
 
 makedocs(;
-    modules=[GraphNeuralNetworks, NNlib],
+    modules=[GraphNeuralNetworks],
     sitename = "GraphNeuralNetworks.jl",
     pages = ["Home" => "index.md",
              "GNNGraph" => "gnngraph.md",

examples/graph_classification_tudataset.jl

@@ -73,11 +73,8 @@ function train(; kws...)
 
     gfull = getdataset()
 
-    @info "MUTAG DATASET
-            num_nodes: $(gfull.num_nodes)  
-            num_edges: $(gfull.num_edges)  
-            num_graphs: $(gfull.num_graphs)"
-
+    @info gfull
+    
     perm = randperm(gfull.num_graphs)
     gtrain = getobs(gfull, perm[1:NUM_TRAIN])
     gtest = getobs(gfull, perm[NUM_TRAIN+1:end]) 

src/gnngraph.jl

@@ -166,6 +166,26 @@ function GNNGraph(g::GNNGraph; ndata=g.ndata, edata=g.edata, gdata=g.gdata)
             ndata, edata, gdata) 
 end
 
+function Base.show(io::IO, g::GNNGraph)
+    println(io, "GNNGraph:
+    num_nodes = $(g.num_nodes)
+    num_edges = $(g.num_edges)
+    num_graphs = $(g.num_graphs)
+    # feature name => array size")
+    println(io, "    ndata:")
+    for k in keys(g.ndata)
+        println(io, "        $k => $(size(g.ndata[k]))")
+    end
+    println(io, "    edata:")
+    for k in keys(g.edata)
+        println(io, "        $k => $(size(g.edata[k]))")
+    end
+    println(io, "    gdata:")
+    for k in keys(g.gdata)
+        println(io, "        $k => $(size(g.gdata[k]))")
+    end
+end
+
 """
     edge_index(g::GNNGraph)