Update src/mlj_embedder_interface.jl

FluxML · Jan 20, 2025 · 9811554 · 9811554
1 parent b879986
commit 9811554
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mlj_embedder_interface.jl b/src/mlj_embedder_interface.jl
@@ -101,7 +101,7 @@ Train the machine using `fit!(mach)`.
 # Hyper-parameters
 
 - `model`: The supervised MLJFlux neural network model to be used for entity embedding. 
-  This must be one of these: `MLJFlux.NeuralNetworkClassifier`, `NeuralNetworkBinaryClassifier`
+  This must be one of these: `MLJFlux.NeuralNetworkClassifier`, `NeuralNetworkBinaryClassifier`,
   `MLJFlux.NeuralNetworkRegressor`,`MLJFlux.MultitargetNeuralNetworkRegressor`. The selected model may have hyperparameters 
   that may affect embedding performance, the most notable of which could be the `builder` argument.