Contents

module EasyML
  module Models
    module Hyperparameters
      class XGBoost < Base
        include GlueGun::DSL

        attribute :learning_rate, :float, default: 0.1
        attribute :max_depth, :integer, default: 6
        attribute :n_estimators, :integer, default: 100
        attribute :booster, :string, default: "gbtree"
        attribute :objective, :string, default: "reg:squarederror"

        validates :objective,
                  inclusion: { in: %w[binary:logistic binary:hinge multi:softmax multi:softprob reg:squarederror
                                      reg:logistic] }
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
easy_ml-0.1.4	lib/easy_ml/core/models/hyperparameters/xgboost.rb
easy_ml-0.1.3	lib/easy_ml/core/models/hyperparameters/xgboost.rb
easy_ml-0.1.2	lib/easy_ml/core/models/hyperparameters/xgboost.rb
easy_ml-0.1.1	lib/easy_ml/core/models/hyperparameters/xgboost.rb