add OGBDataset (#106)

Author: CarloLucibello

README.md

@@ -38,11 +38,13 @@ Find below a list of available datasets and links to their documentation.
   - [UD_English](https://juliaml.github.io/MLDatasets.jl/dev/datasets/UD_English/)
 
 #### Graphs
-  - [CiteSeer](https://juliaml.github.io/MLDatasets.jl/dev/datasets/CiteSeer/)
-  - [Cora](https://juliaml.github.io/MLDatasets.jl/dev/datasets/Cora/)
-  - [PubMed](https://juliaml.github.io/MLDatasets.jl/dev/datasets/PubMed/)
-  - [TUDatasets](https://juliaml.github.io/MLDatasets.jl/dev/datasets/TUDataset/)
 
+Documentation [link](https://juliaml.github.io/MLDatasets.jl/dev/datasets/graphs). Available datasets:
+  - CiteSeer
+  - Cora
+  - OGBDataset
+  - PubMed
+  - TUDataset
 
 ## Installation
 

docs/make.jl

@@ -42,12 +42,7 @@ makedocs(
                 "UD_English" => "datasets/UD_English.md",
             ],
 
-            "Graphs" => Any[
-                "CiteSeer" => "datasets/CiteSeer.md",
-                "Cora" => "datasets/Cora.md",
-                "PubMed" => "datasets/PubMed.md",
-                "TUDataset" => "datasets/TUDataset.md",      
-            ],
+            "Graphs" => "datasets/graphs.md",
 
         ],
         "Utils" => "utils.md",

docs/src/datasets/CiteSeer.md

@@ -0,0 +1,16 @@
+# Graph Datasets
+
+```@index
+Pages = ["graphs.md"]
+```
+
+```@docs
+CiteSeer
+CiteSeer.dataset
+Cora
+Cora.dataset
+OGBDataset
+PubMed
+PubMed.dataset
+TUDataset
+```

docs/src/datasets/Cora.md

@@ -17,32 +17,7 @@ import MLUtils
 using MLUtils: getobs, numobs, AbstractDataContainer
 export getobs, numobs
 
-# Julia 1.0 compatibility
-if !isdefined(Base, :isnothing)
-    isnothing(x) = x === nothing
-end
-
-bytes_to_type(::Type{UInt8}, A::Array{UInt8}) = A
-bytes_to_type(::Type{N0f8}, A::Array{UInt8}) = reinterpret(N0f8, A)
-bytes_to_type(::Type{T}, A::Array{UInt8}) where T<:Integer = convert(Array{T}, A)
-bytes_to_type(::Type{T}, A::Array{UInt8}) where T<:AbstractFloat = A ./ T(255)
-bytes_to_type(::Type{T}, A::Array{UInt8}) where T<:Number  = convert(Array{T}, reinterpret(N0f8, A))
-
-global __images_supported__ = false
-
-function _channelview(image::AbstractArray{<:Colorant})
-    __images_supported__ ||
-        error("Converting to/from image requires `using ImageCore`")
-    ImageCore.channelview(image)
-end
-
-function _colorview(::Type{T}, array::AbstractArray{<:Number}) where T <: Color
-    __images_supported__ ||
-        error("Converting to image requires `using ImageCore`")
-    ImageCore.colorview(T, array)
-end
-
-
+include("utils.jl")
 include("download.jl")
 
 include("containers/filedataset.jl")
@@ -56,6 +31,8 @@ export JLD2Dataset
 include("containers/cacheddataset.jl")
 export CachedDataset
 
+include("abstract_dataset.jl")
+
 # Misc.
 include("BostonHousing/BostonHousing.jl")
 include("Iris/Iris.jl")
@@ -81,6 +58,8 @@ include("planetoid.jl")
     include("PubMed/PubMed.jl")
     include("CiteSeer/CiteSeer.jl")
 include("TUDataset/TUDataset.jl")
+include("OGBDataset/OGBDataset.jl")
+
 
 function __init__()
     # initialize optional dependencies
@@ -89,6 +68,7 @@ function __init__()
     end
 
     __init__tudataset()
+    __init__ogbdataset()
 end
 
 end