Sha256: 65de86803e4f2c9b6a8555ab23a41260a0c0a45e49e3924af39a0d068b8313f7

Contents?: true

Size: 787 Bytes

Versions: 88

Compression:

Stored size: 787 Bytes

Contents

module EasyML
  module Models
    class XGBoost
      class ProgressCallback < ::XGBoost::TrainingCallback
        def initialize(options)
        end

        def as_json
          { callback_type: :progress_callback }
        end

        def before_iteration(*args)
          false
        end

        attr_reader :model

        def model=(model)
          @model = model
        end

        def after_iteration(booster, epoch, history)
          if model.adapter.progress_callback
            model.adapter.progress_callback.call({ iteration: epoch, evals: history })
          end
          return false
        end

        def before_training(booster)
          booster
        end

        def after_training(booster)
          booster
        end
      end
    end
  end
end

Version data entries

88 entries across 88 versions & 1 rubygems

Version Path
easy_ml-0.2.0.pre.rc101 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc100 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc99 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc98 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc97 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc96 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc95 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc94 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc93 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc92 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc91 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc90 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc89 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc88 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc85 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc84 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc83 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc82 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc81 app/models/easy_ml/models/xgboost/progress_callback.rb
easy_ml-0.2.0.pre.rc78 app/models/easy_ml/models/xgboost/progress_callback.rb