Adicionando VirtualDrift

Author: filipebraida

src/modifiers.jl

@@ -38,7 +38,6 @@ FilterModifier(columns::Symbol...) = FilterModifier([columns...])
 function apply!(modifier::FilterModifier, data::DataFrame, event::Int)
     columns = Symbol[]
     for col in modifier.columns
-        #TODO: Não ficou boa essa checagem.
         if !(col in propertynames(data))
             #TODO: Colocar para avisar somente uma única vez do problema.
             @warn "O stream não possui a $col"
@@ -50,6 +49,7 @@ function apply!(modifier::FilterModifier, data::DataFrame, event::Int)
     select!(data, columns)
     return nothing
 end
+
 struct DriftModifier <: Modifier
     column::Symbol
     filter::Function
@@ -61,4 +61,16 @@ function apply!(modifier::DriftModifier, data::DataFrame, event::Int)
     data[elements, modifier.column] = data[elements, modifier.column] .+ modifier.drift(event)
 
     return nothing
+end
+
+sigmoid(x; c1 = 1.0, c2 = 0.0) = 1 / (1 + ℯ ^ (-c1 * (x - c2)))
+
+function IncrementalVirtualDriftModifier(vetor::Dict{Symbol, T}, filter::Function; c1 = 1.0, c2 = 0.0)::Modifiers where T <: Number
+    modifiers = EasyStream.Modifier[]
+    for (column, value) in vetor
+        drift = DriftModifier(column, filter, x -> value .* sigmoid(x; c1 = c1, c2 = c2))
+        push!(modifiers, drift)
+    end
+
+    return Modifiers(modifiers)
 end