Method: EasyML::Data::Preprocessor::SimpleImputer#apply_defaults

Defined in:
lib/easy_ml/data/preprocessor/simple_imputer.rb

#apply_defaults ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/easy_ml/data/preprocessor/simple_imputer.rb', line 30

def apply_defaults
  @options[:date_column] ||= "CREATED_DATE"

  if strategy == :categorical
    @options[:categorical_min] ||= 25
  elsif strategy == :custom
    itself = ->(col) { col }
    @options[:fit] ||= itself
    @options[:transform] ||= itself
  end
end