Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jdb/api #276

Open
wants to merge 19 commits into
base: main
Choose a base branch
from
Open

Jdb/api #276

wants to merge 19 commits into from

Conversation

jeremiedb
Copy link
Member

@jeremiedb jeremiedb commented Jan 27, 2025

Breaking changes:

Model constructors (EvoTreeRegressor, EvoTreeClassifier...) now include the following arguments:
  • device
  • early_stopping_rounds
  • metric

Example:

config = EvoTreeRegressor(; loss=:mse, device=:gpu, L2=100)

Change in naming of variables identity in the Tables / DataFrames based internal API, which were previously kwargs of fit:

  • fnames => feature_names
  • w_name => weight_names
m = fit_evotree(config, dtrain; target_name="y", feature_names=["x1", "x2"])

Changes related to losses:

  • L1 / l1 loss is no longer supported. Use loss=:mae in EvoTreeRegressor instead.

Fixes and improvements to GPU:

  • Models trained through MLJ now support the :gpu argument.
  • Inference is now properly dispatch to :gpu when using: m(dtrain; device=:gpu)
  • Both :mae and :quantile losses are now now suported on GPU (device = :gpu)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant