search

IQVIA-ML / TreeParzen.jl

TreeParzen.jl, a pure Julia hyperparameter optimiser with MLJ integration
Other
35 stars 5 forks source link

Update MLJTuning.jl example #101

Open ParadaCarleton opened 11 months ago

ParadaCarleton commented 11 months ago

For example, the MLJTuning.jl currently references an undefined HP package (?) and uses unusual ranges:

space = Dict(
    :num_round => HP.QuantUniform(:num_round, 1., 500., 1.),
    :eta => HP.LogUniform(:eta, -3., 0.),
    :gamma => HP.LogUniform(:gamma, -3., 3.),
    :max_depth => HP.QuantUniform(:max_depth, 1., ceil(log2(training_data_per_fold)), 1.0),
    :min_child_weight => HP.LogUniform(:min_child_weight, -5., 2.),
    :lambda => HP.LogUniform(:lambda, -5., 2.),
    :alpha => HP.LogUniform(:alpha, -5., 2.),
)
kainkad commented 6 months ago

Hi @ParadaCarleton HP is an internal TreeParzen module containing all of the listed above functions. I was wondering if you meant using the range() as mentioned in the #100?