classif fixes

api cleanup
Evovest · Apr 21, 2024 · 40d2f75 · 40d2f75
1 parent a4f8dc7
commit 40d2f75
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
+authors = ["jeremiedb <[email protected]>"]
 name = "NeuroTreeModels"
 uuid = "1db4e0a5-a364-4b0c-897c-2bd5a4a3a1f2"
-authors = ["jeremiedb <[email protected]>"]
 version = "1.3.0"
 
 [deps]
@@ -20,6 +20,7 @@ cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
 [compat]
 CUDA = "4, 5"
+CategoricalArrays = "0.10"
 ChainRulesCore = "1"
 DataFrames = "1.3"
 Flux = "0.13, 0.14"
@@ -34,11 +35,10 @@ cuDNN = "1"
 julia = "1.9"
 
 [extras]
-CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
 MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458"
 MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
 MLJTestInterface = "72560011-54dd-4dc2-94f3-c5de45b75ecd"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test", "CategoricalArrays", "MLDatasets", "MLJTestInterface", "MLJBase"]
+test = ["Test", "MLDatasets", "MLJTestInterface", "MLJBase"]
diff --git a/README.md b/README.md
@@ -25,7 +25,9 @@
 
 ## Configuring a model
 
-A model configuration is defined with the [NeuroTreeRegressor](https://evovest.github.io/NeuroTreeModels.jl/dev/models#NeuroTreeModels.NeuroTreeRegressor) constructor:
+A model configuration is defined with on of the constructor:
+- [NeuroTreeRegressor](https://evovest.github.io/NeuroTreeModels.jl/dev/models#NeuroTreeModels.NeuroTreeRegressor)
+- [NeuroTreeClassifier](https://evovest.github.io/NeuroTreeModels.jl/dev/models#NeuroTreeModels.NeuroTreeClassifier)
 
 ```julia
 using NeuroTreeModels, DataFrames

diff --git a/docs/src/models.md b/docs/src/models.md
@@ -1,10 +1,18 @@
 # Models
 
+```@contents
+```
+
 ## NeuroTreeRegressor
 ```@docs
 NeuroTreeRegressor
 ```
 
+## NeuroTreeClassifier
+```@docs
+NeuroTreeClassifier
+```
+
 ## NeuroTreeModel
 ```@docs
 NeuroTreeModel