From 4547f2a77197d5e2f089c3a2704bc8d988139ec2 Mon Sep 17 00:00:00 2001 From: Essam Date: Sat, 31 Aug 2024 18:51:27 -0500 Subject: [PATCH] Update src/classifier.jl Co-authored-by: Anthony Blaom, PhD --- src/classifier.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/classifier.jl b/src/classifier.jl index d3889e5..b08c996 100644 --- a/src/classifier.jl +++ b/src/classifier.jl @@ -47,7 +47,7 @@ end MLJModelInterface.metadata_model( NeuralNetworkClassifier, - input_scitype = Union{AbstractMatrix{Continuous}, Table(Finite)}, + input_scitype = Union{AbstractMatrix{Continuous}, Table(Continuous,Finite)}, target_scitype = AbstractVector{<:Finite}, load_path = "MLJFlux.NeuralNetworkClassifier", )