From cf30f9033eb2b7e959c0d8e1279d7199caa07da7 Mon Sep 17 00:00:00 2001 From: Essam Date: Mon, 5 Aug 2024 18:31:34 -0500 Subject: [PATCH] =?UTF-8?q?=E2=9E=95=20Include=20necessary=20files=20in=20?= =?UTF-8?q?MLJFlux.jl?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/MLJFlux.jl | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/src/MLJFlux.jl b/src/MLJFlux.jl index 9a2ed87..932b551 100644 --- a/src/MLJFlux.jl +++ b/src/MLJFlux.jl @@ -2,7 +2,6 @@ module MLJFlux export CUDALibs, CPU1 -import Flux using MLJModelInterface using MLJModelInterface.ScientificTypesBase using ScientificTypes: schema, Finite @@ -14,12 +13,12 @@ using Statistics using ColorTypes using ComputationalResources using Random -import Metalhead -import Optimisers include("utilities.jl") -const MMI=MLJModelInterface +const MMI = MLJModelInterface +include("encoders.jl") +include("entity_embedding.jl") include("builders.jl") include("metalhead.jl") include("types.jl") @@ -28,15 +27,11 @@ include("regressor.jl") include("classifier.jl") include("image.jl") include("mlj_model_interface.jl") -include("entity_embedding.jl") -include("ordinal_encoder.jl") export NeuralNetworkRegressor, MultitargetNeuralNetworkRegressor export NeuralNetworkClassifier, NeuralNetworkBinaryClassifier, ImageClassifier export CUDALibs, CPU1 -export CategoricalEmbedder +export EntityEmbedder include("deprecated.jl") - - end