Skip to content

Commit

Permalink
use Nx.BinaryBackend for cv notebook (elixir-nx#274)
Browse files Browse the repository at this point in the history
  • Loading branch information
santiago-imelio authored Jun 1, 2024
1 parent 6fdf3a7 commit 32a5b56
Showing 1 changed file with 3 additions and 11 deletions.
14 changes: 3 additions & 11 deletions notebooks/cv_gradient_boosting_tree.livemd
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,13 @@
```elixir
Mix.install([
{:scholar, "~> 0.2.0"},
{:explorer, "~> 0.6.1"},
{:exla, "~> 0.6.0"},
{:nx, "~> 0.6.0", override: true},
{:exgboost, "~> 0.3"},
{:req, "~> 0.3.9"},
{:kino_vega_lite, "~> 0.1.9"},
{:kino, "~> 0.10.0"},
{:kino_explorer, "~> 0.1.7"}
{:kino_explorer, "~> 0.1.7"},
{:nx, "~> 0.7.2"},
{:explorer, "~> 0.8.2"}
])
```

Expand All @@ -23,13 +22,6 @@ 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 32a5b56

Please sign in to comment.