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