Noob question. Can I use KotlinDL for text classif...
# kotlindl
a
Noob question. Can I use KotlinDL for text classification?? If yes, any example or doc
j
We have not really explored text classification, so there are no examples. What kind of classification model do you want to implement?
a
@Julia Beliaeva multiple category classification. For example bank transaction classification
j
Do you have an architecture in mind that you'd like to use?
a
Embedding layer is not supported yet!☹️
@Julia Beliaeva, is it in the roadmap??
j
We don't have any specific plans for it right now.
a
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