Implement the 'parameters' function for some transforms

Author: eliascarv

Project.toml

@@ -38,6 +38,6 @@ Statistics = "1.9"
 StatsBase = "0.33, 0.34"
 Tables = "1.6"
 Transducers = "0.4"
-TransformsBase = "1.3"
+TransformsBase = "1.4"
 Unitful = "1.17"
 julia = "1.9"

src/TableTransforms.jl

@@ -27,7 +27,7 @@ using Transducers: tcollect
 using NelderMead: optimise
 
 import Distributions: quantile, cdf
-import TransformsBase: assertions, isrevertible, isinvertible
+import TransformsBase: assertions, parameters, isrevertible, isinvertible
 import TransformsBase: apply, revert, reapply, preprocess, inverse
 
 include("tabletraits.jl")

src/transforms/coalesce.jl

@@ -42,6 +42,8 @@ Coalesce(; value) = Coalesce(AllSelector(), value)
 Coalesce(cols; value) = Coalesce(selector(cols), value)
 Coalesce(cols::C...; value) where {C<:Column} = Coalesce(selector(cols), value)
 
+parameters(transform::Coalesce) = (; value=transform.value)
+
 isrevertible(::Type{<:Coalesce}) = false
 
 colcache(::Coalesce, x) = nothing

src/transforms/dropextrema.jl

@@ -50,6 +50,8 @@ DropExtrema(; low=0.25, high=0.75) = DropExtrema(AllSelector(), low, high)
 DropExtrema(cols; low=0.25, high=0.75) = DropExtrema(selector(cols), low, high)
 DropExtrema(cols::C...; low=0.25, high=0.75) where {C<:Column} = DropExtrema(selector(cols), low, high)
 
+parameters(transform::DropExtrema) = (low=transform.low, high=transform.high)
+
 isrevertible(::Type{<:DropExtrema}) = false
 
 function preprocess(transform::DropExtrema, feat)

src/transforms/eigenanalysis.jl

@@ -54,6 +54,8 @@ EigenAnalysis(proj; maxdim=nothing, pratio=1.0) = EigenAnalysis(proj, maxdim, pr
 
 assertions(::EigenAnalysis) = [scitypeassert(Continuous)]
 
+parameters(transform::EigenAnalysis) = (proj=transform.proj, maxdim=transform.maxdim, pratio=transform.pratio)
+
 isrevertible(::Type{EigenAnalysis}) = true
 
 function applyfeat(transform::EigenAnalysis, feat, prep)

src/transforms/indicator.jl

@@ -48,6 +48,8 @@ Indicator(col::Column; k=10, scale=:quantile, categ=false) = Indicator(selector(
 
 assertions(transform::Indicator) = [scitypeassert(Continuous, transform.selector)]
 
+parameters(transform::Indicator) = (k=transform.k, scale=transform.scale)
+
 isrevertible(::Type{<:Indicator}) = true
 
 function _intervals(transform::Indicator, x)

src/transforms/projectionpursuit.jl

@@ -49,6 +49,8 @@ ProjectionPursuit(; tol=1e-6, maxiter=100, deg=5, perc=0.9, n=100, rng=Random.GL
 
 assertions(::ProjectionPursuit) = [scitypeassert(Continuous)]
 
+parameters(transform::ProjectionPursuit) = (tol=transform.tol, deg=transform.deg, perc=transform.perc, n=transform.n)
+
 isrevertible(::Type{<:ProjectionPursuit}) = true
 
 # transforms a row of random variables into a convex combination 

src/transforms/quantile.jl

@@ -41,6 +41,8 @@ Quantile(cols::C...; dist=Normal()) where {C<:Column} = Quantile(selector(cols),
 
 assertions(transform::Quantile) = [scitypeassert(Continuous, transform.selector)]
 
+parameters(transform::Quantile) = (; dist=transform.dist)
+
 isrevertible(::Type{<:Quantile}) = true
 
 colcache(::Quantile, x) = EmpiricalDistribution(x)

src/transforms/remainder.jl

@@ -22,6 +22,8 @@ isrevertible(::Type{<:Remainder}) = true
 
 assertions(::Remainder) = [scitypeassert(Continuous)]
 
+parameters(transform::Remainder) = (; total=transform.total)
+
 function applyfeat(transform::Remainder, feat, prep)
   cols = Tables.columns(feat)
   names = Tables.columnnames(cols) |> collect

src/transforms/scale.jl

@@ -50,6 +50,8 @@ Scale(cols::C...; low=0.25, high=0.75) where {C<:Column} = Scale(selector(cols),
 
 assertions(transform::Scale) = [scitypeassert(Continuous, transform.selector)]
 
+parameters(transform::Scale) = (low=transform.low, high=transform.high)
+
 isrevertible(::Type{<:Scale}) = true
 
 function colcache(transform::Scale, x)