Don't use tracked values here

FluxML · May 4, 2020 · c77f405 · c77f405
1 parent 344ebe0
commit c77f405
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/regressor.jl b/src/regressor.jl
@@ -143,7 +143,7 @@ function MLJModelInterface.predict(model::Regressor, fitresult, Xnew_)
         return MLJModelInterface.table(reduce(hcat, y for y in ypred)',
                                        names=target_column_names)
     else
-        return [chain(values.(Xnew_[i, :]))[1] for i in 1:size(Xnew_, 1)]
+        return [chain(values.(Xnew_[i, :]))[1].data for i in 1:size(Xnew_, 1)]
     end
 end