Example of using Embedding about ruby-dnn HOT 3 OPEN

Thanks for your response. I will try that. On Apr 7, 2020, at 6:46 AM, unagiootoro <[email protected]> wrote:  Hi, Thank you for use to this library. Currently, ruby-dnn does not support Slice. However, by giving multiple inputs to the model, the it is possible to apply different embeddings to two different categorical features. It can be accomplished with the following code. class MyModel < Model ##... def forward(inputs) x1, x2 = *inputs e1 = @embed1.(x1) e2 = @embed2.(x2) x = Concatenate.(e1, e2, axis: 1) x = @Dense.(x) endend model = MyModel.new model.train([input1, input2], output, epochs) And, Embedding can not be applied to multi-dimensional features. Therefore, after applying Embedding to a one-dimensional feature, the you need to make it multi-dimensional with Reshape. class MyModel < Model ##... def forward(x) e = @embed.(x) x = Reshape.(x, [32, 32, 3]) x = @conv2d.(x) endend ※I am not good at English and may be wrong. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#10 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AIC2TXLIOTWIFTPHFH55K73RLMVE3ANCNFSM4L64H7BA> .