Update src/regressor.jl

Co-authored-by: Anthony Blaom, PhD <[email protected]>
FluxML · Aug 31, 2024 · 9a836b2 · 9a836b2
1 parent e8e87cd
commit 9a836b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/regressor.jl b/src/regressor.jl
@@ -80,6 +80,6 @@ function MLJModelInterface.predict(model::MultitargetNeuralNetworkRegressor,
 end
 
 MLJModelInterface.metadata_model(MultitargetNeuralNetworkRegressor,
-    input = Union{AbstractMatrix{Continuous}, Table(Finite)},
+    input = Union{AbstractMatrix{Continuous}, Table(Continuous,Finite)},
     target = Union{AbstractMatrix{Continuous}, Table(Continuous)},
     path = "MLJFlux.MultitargetNeuralNetworkRegressor")