Skip to content

Commit

Permalink
Merge branch 'main' of github.com:acalejos/exgboost
Browse files Browse the repository at this point in the history
  • Loading branch information
acalejos committed Feb 19, 2024
2 parents 380c1a0 + 893dd5e commit 05d15f0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ primary entry point for training a model. It accepts a Nx tensor for the feature
accepts a keyword list of options that can be used to configure the training process. See the
[XGBoost documentation](https://xgboost.readthedocs.io/en/latest/parameter.html) for the full list of options.

`Exgbost.train/2` uses the `EXGBoost.Training.train/1` function to perform the actual training. `EXGBoost.Training.train/1`
`EXGBoost.train/2` uses the `EXGBoost.Training.train/1` function to perform the actual training. `EXGBoost.Training.train/1`
and can be used directly if you wish to work directly with the `DMatrix` and `Booster` structs.

One of the main features of `EXGBoost.train/2` is the ability for the end user to provide a custom training function
Expand Down Expand Up @@ -96,7 +96,7 @@ to `EXGBoost.train/2`. See the [XGBoost documentation](https://xgboost.readthedo
list of parameters.

```elixir
Exgboot.train(X,
EXGBoost.train(X,
y,
obj: &EXGBoost.Training.train/1,
evals: [{X_test, y_test, "test"}],
Expand Down

0 comments on commit 05d15f0

Please sign in to comment.