Update src/classifier.jl

Co-authored-by: Anthony Blaom, PhD <[email protected]>
FluxML · Aug 31, 2024 · 4547f2a · 4547f2a
1 parent c90cf6e
commit 4547f2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",
 )