add augmented_transform interface

OutlierDetectionJL · Oct 15, 2021 · 1d8f2db · 1d8f2db
1 parent 57b2141
commit 1d8f2db
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/interface.jl b/src/interface.jl
@@ -21,6 +21,12 @@ function MMI.transform(detector::Detector, model::DetectorModel, X)
     return scores
 end
 
+function augmented_transform(detector::Detector, fitresult::Fit, X)
+    model, scores_train = fitresult
+    scores_test = MMI.transform(detector, model, X)
+    return scores_train, scores_test
+end
+
 # specify scitypes
 MMI.input_scitype(::Type{<:Detector}) = Union{AbstractMatrix{<:MMI.Continuous}, MMI.Table(MMI.Continuous)}
 MMI.target_scitype(::Type{<:Detector}) = AbstractVector{<:Union{Missing,OrderedFactor{2}}}