🔼 Improve TargetTransform interface

EssamWisam · EssamWisam · commit 8212fd517868 · 2024-07-22T22:05:21.000-05:00
diff --git a/src/encoders/target_encoding/interface_mlj.jl b/src/encoders/target_encoding/interface_mlj.jl
@@ -111,12 +111,7 @@ MMI.metadata_model(
     load_path = "MLJTransforms.TargetEncoder",
 )
 
-function MMI.fit_data_scitype(t::TargetEncoder)
-    return Tuple{
-        Table,
-        AbstractVector,
-    }
-end
+MMI.target_in_fit(::Type{<:TargetEncoder}) = true
 
 
 """
