Reorder in alphabetical order for consistency (#157)

Author: christiangnrd

docs/src/datasets/graphs.md

@@ -23,10 +23,10 @@ MLDatasets.HeteroGraph
 CiteSeer
 Cora
 KarateClub
+MovieLens
 OGBDataset
 PolBlogs
 PubMed
 Reddit
 TUDataset
-MovieLens
 ```

src/MLDatasets.jl

@@ -71,23 +71,24 @@ export Titanic
 
 ## Vision
 
-include("datasets/vision/emnist.jl")
-export EMNIST
-include("datasets/vision/mnist_reader/MNISTReader.jl")
-include("datasets/vision/mnist.jl")
-export MNIST
-include("datasets/vision/fashion_mnist.jl")
-export FashionMNIST
 include("datasets/vision/cifar10_reader/CIFAR10Reader.jl")
 include("datasets/vision/cifar10.jl")
 export CIFAR10
 include("datasets/vision/cifar100_reader/CIFAR100Reader.jl")
 include("datasets/vision/cifar100.jl")
 export CIFAR100
-include("datasets/vision/svhn2.jl")
-export SVHN2
+include("datasets/vision/emnist.jl")
+export EMNIST
+include("datasets/vision/fashion_mnist.jl")
+export FashionMNIST
+include("datasets/vision/mnist_reader/MNISTReader.jl")
+include("datasets/vision/mnist.jl")
+export MNIST
 include("datasets/vision/omniglot.jl")
 export Omniglot
+include("datasets/vision/svhn2.jl")
+export SVHN2
+
 
 ## Text
 
@@ -106,35 +107,35 @@ include("datasets/graphs/planetoid.jl")
 # export read_planetoid_data
 include("datasets/graphs/cora.jl")
 export Cora
-include("datasets/graphs/pubmed.jl")
-export PubMed
 include("datasets/graphs/citeseer.jl")
 export CiteSeer
-include("datasets/graphs/tudataset.jl")
-export TUDataset
+include("datasets/graphs/karateclub.jl")
+export KarateClub
+include("datasets/graphs/movielens.jl")
+export MovieLens
 include("datasets/graphs/ogbdataset.jl")
 export OGBDataset
 include("datasets/graphs/polblogs.jl")
 export PolBlogs
-include("datasets/graphs/karateclub.jl")
-export KarateClub
+include("datasets/graphs/pubmed.jl")
+export PubMed
 include("datasets/graphs/reddit.jl")
 export Reddit
-include("datasets/graphs/movielens.jl")
-export MovieLens
+include("datasets/graphs/tudataset.jl")
+export TUDataset
 
 function __init__()
     # TODO automatically find and execute __init__xxx functions
 
     # graph
     __init__citeseer()
     __init__cora()
+    __init__movielens()
     __init__ogbdataset()
     __init__polblogs()
     __init__pubmed()
     __init__reddit()
     __init__tudataset()
-    __init__movielens()
 
     # misc
     __init__iris()

test/datasets/graphs.jl

@@ -1,29 +1,4 @@
 
-@testset "Cora" begin
-    data  = Cora()
-    @test data isa AbstractDataset
-    @test length(data) == 1
-    g = data[1]
-    @test g === data[:]
-    @test g isa MLDatasets.Graph
-
-    @test g.num_nodes == 2708
-    @test g.num_edges == 10556
-    @test size(g.node_data.features) == (1433, g.num_nodes)
-    @test size(g.node_data.targets) == (g.num_nodes,)
-    @test sum(g.node_data.train_mask) == 140
-    @test sum(g.node_data.val_mask) == 500
-    @test sum(g.node_data.test_mask) == 1000
-    @test g.edge_index isa Tuple{Vector{Int}, Vector{Int}}
-    s, t = g.edge_index
-    for a in (s, t)
-        @test a isa Vector{Int}
-        @test length(a) == g.num_edges
-        @test minimum(a) == 1
-        @test maximum(a) == g.num_nodes
-    end
-end
-
 @testset "CiteSeer" begin
     data  = CiteSeer()
     @test data isa AbstractDataset
@@ -49,19 +24,19 @@ end
     end
 end
 
-@testset "PubMed" begin
-    data  = PubMed()
+@testset "Cora" begin
+    data  = Cora()
     @test data isa AbstractDataset
     @test length(data) == 1
     g = data[1]
     @test g === data[:]
     @test g isa MLDatasets.Graph
 
-    @test g.num_nodes == 19717
-    @test g.num_edges == 88648
-    @test size(g.node_data.features) == (500, g.num_nodes)
+    @test g.num_nodes == 2708
+    @test g.num_edges == 10556
+    @test size(g.node_data.features) == (1433, g.num_nodes)
     @test size(g.node_data.targets) == (g.num_nodes,)
-    @test sum(g.node_data.train_mask) == 60
+    @test sum(g.node_data.train_mask) == 140
     @test sum(g.node_data.val_mask) == 500
     @test sum(g.node_data.test_mask) == 1000
     @test g.edge_index isa Tuple{Vector{Int}, Vector{Int}}
@@ -99,7 +74,6 @@ end
     end
 end
 
-
 @testset "PolBlogs" begin
     data  = PolBlogs()
     @test data isa AbstractDataset
@@ -123,6 +97,31 @@ end
     end
 end
 
+@testset "PubMed" begin
+    data  = PubMed()
+    @test data isa AbstractDataset
+    @test length(data) == 1
+    g = data[1]
+    @test g === data[:]
+    @test g isa MLDatasets.Graph
+
+    @test g.num_nodes == 19717
+    @test g.num_edges == 88648
+    @test size(g.node_data.features) == (500, g.num_nodes)
+    @test size(g.node_data.targets) == (g.num_nodes,)
+    @test sum(g.node_data.train_mask) == 60
+    @test sum(g.node_data.val_mask) == 500
+    @test sum(g.node_data.test_mask) == 1000
+    @test g.edge_index isa Tuple{Vector{Int}, Vector{Int}}
+    s, t = g.edge_index
+    for a in (s, t)
+        @test a isa Vector{Int}
+        @test length(a) == g.num_edges
+        @test minimum(a) == 1
+        @test maximum(a) == g.num_nodes
+    end
+end
+
 # maybe, maybe, maybe??
 Sys.iswindows() || @testset "OGBn-mag" begin
     data = OGBDataset("ogbn-mag")