From 9a836b20f58fcb2e8072decb3ca27231d920721f Mon Sep 17 00:00:00 2001 From: Essam Date: Sat, 31 Aug 2024 18:52:11 -0500 Subject: [PATCH] Update src/regressor.jl Co-authored-by: Anthony Blaom, PhD --- src/regressor.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/regressor.jl b/src/regressor.jl index c343608..e55acbf 100644 --- 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")