Skip to content

Commit d95fc8e

Browse files
EssamWisamablaom
andauthored
Update src/mlj_embedder_interface.jl
Co-authored-by: Anthony Blaom, PhD <anthony.blaom@gmail.com>
1 parent 757c186 commit d95fc8e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/mlj_embedder_interface.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ Train the machine using `fit!(mach)`.
7575
7676
# Hyper-parameters
7777
78-
- `model`: The underlying deep learning model to be used for entity embedding. So far this supports `NeuralNetworkClassifier`, `NeuralNetworkRegressor`, and `MultitargetNeuralNetworkRegressor`.
78+
- `model`: The supervised MLJFlux neural network model to be used for entity embedding.
79+
This must be one of these: `MLJFlux.NeuralNetworkClassifier`,
80+
`MLJFlux.NeuralNetworkRegressor`,`MLJFlux.MultitargetNeuralNetworkRegressor`.
7981
8082
# Operations
8183

0 commit comments

Comments
 (0)