https://kotlinlang.org logo
Title
a

ayodele

08/22/2022, 9:31 PM
Noob question. Can I use KotlinDL for text classification?? If yes, any example or doc
j

Julia Beliaeva

08/24/2022, 8:14 PM
We have not really explored text classification, so there are no examples. What kind of classification model do you want to implement?
a

ayodele

08/24/2022, 9:08 PM
@Julia Beliaeva multiple category classification. For example bank transaction classification
j

Julia Beliaeva

08/24/2022, 9:39 PM
Do you have an architecture in mind that you'd like to use?
a

ayodele

08/30/2022, 2:18 PM
Embedding layer is not supported yet!☹️
@Julia Beliaeva, is it in the roadmap??
j

Julia Beliaeva

09/01/2022, 11:24 AM
We don't have any specific plans for it right now.
a

Ananiya

09/12/2022, 10:04 AM
You could use library djl, or londogard nlp to embed the word indices before its fed to the NN however limitations like absence of RNN or MHA layers would lead to lower accuracy so IMO the best method could be loading pretrained onnx model through
OnnxInferenceModel
(Provided by kdl) from HuggingFace