diff --git a/README.md b/README.md index b0399d7..12051e1 100644 --- a/README.md +++ b/README.md @@ -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 @@ -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"}],