Sha256: 65de86803e4f2c9b6a8555ab23a41260a0c0a45e49e3924af39a0d068b8313f7
Contents?: true
Size: 787 Bytes
Versions: 92
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
92 entries across 92 versions & 1 rubygems