Skip to content

Commit 9de68f3

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

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/mlj_embedder_interface.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ X = (;
107107
y = categorical([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) # Classification
108108
109109
# Initiate model
110+
EntityEmbedder = @load EntityEmbedder pkg=MLJFlux
110111
NeuralNetworkClassifier = @load NeuralNetworkClassifier pkg=MLJFlux
111112
112113
clf = NeuralNetworkClassifier(embedding_dims=Dict(:Column2 => 2, :Column3 => 2))

0 commit comments

Comments
 (0)