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", )