From 5accc3e20e4c000c370878b63aa737e864a4f0ea Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Thu, 25 Jul 2024 16:04:50 +0200 Subject: [PATCH] mlutils --- Project.toml | 5 +++-- src/GraphNeuralNetworks.jl | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 5506b943a..a8bdcf02f 100644 --- a/Project.toml +++ b/Project.toml @@ -11,6 +11,7 @@ Functors = "d9f16b24-f501-4c13-a1f2-28368ffc5196" GNNGraphs = "aed8fd31-079b-4b5a-b342-a13352159b8c" LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e" MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09" +MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54" NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c" Reexport = "189a3867-3050-52da-a836-e630ba90ab69" @@ -31,6 +32,7 @@ Functors = "0.4.1" GNNGraphs = "1.0" LinearAlgebra = "1" MacroTools = "0.5" +MLUtils = "0.4" NNlib = "0.9" Random = "1" Reexport = "1" @@ -47,11 +49,10 @@ FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000" Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6" InlineStrings = "842dd82b-1e85-43dc-bf29-5d0ee9dffc48" MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458" -MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54" SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf" Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd" [targets] -test = ["Test", "MLUtils", "Adapt", "DataFrames", "InlineStrings", "SparseArrays", "Graphs", "Zygote", "FiniteDifferences", "ChainRulesTestUtils", "MLDatasets", "CUDA", "cuDNN"] +test = ["Test", "Adapt", "DataFrames", "InlineStrings", "SparseArrays", "Graphs", "Zygote", "FiniteDifferences", "ChainRulesTestUtils", "MLDatasets", "CUDA", "cuDNN"] diff --git a/src/GraphNeuralNetworks.jl b/src/GraphNeuralNetworks.jl index 0debce93e..c74032273 100644 --- a/src/GraphNeuralNetworks.jl +++ b/src/GraphNeuralNetworks.jl @@ -10,6 +10,7 @@ using NNlib: scatter, gather using ChainRulesCore using Reexport using DataStructures: nlargest +using MLUtils: zeros_like @reexport using GNNGraphs using GNNGraphs: COO_T, ADJMAT_T, SPARSE_T,