Skip to content

Commit

Permalink
Update EXGBoost version in notebook (#279)
Browse files Browse the repository at this point in the history
  • Loading branch information
acalejos authored Jun 8, 2024
1 parent 341301b commit 09c5ac6
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion notebooks/cv_gradient_boosting_tree.livemd
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
```elixir
Mix.install([
{:scholar, "~> 0.2.0"},
{:exgboost, "~> 0.3"},
{:exgboost, "~> 0.5"},
{:req, "~> 0.3.9"},
{:kino_vega_lite, "~> 0.1.9"},
{:kino, "~> 0.10.0"},
Expand All @@ -22,6 +22,13 @@ require Explorer.DataFrame, as: DF
require Explorer.Series, as: S
```

And let's configure `EXLA` as our default backend (where our tensors are stored) and compiler (which compiles Scholar code) across the notebook and all branched sections:

```elixir
Nx.global_default_backend(EXLA.Backend)
Nx.Defn.global_default_options(compiler: EXLA)
```

We are going to work with [Medical Cost Personal Datasets](https://www.kaggle.com/datasets/mirichoi0218/insurance) to predict medical charges that were applied to each person from the dataset. Let's download it:

```elixir
Expand Down

0 comments on commit 09c5ac6

Please sign in to comment.