search

opensearch-project / ml-commons

ml-commons provides a set of common machine learning algorithms, e.g. k-means, or linear regression, to help developers build ML related features within OpenSearch.
Apache License 2.0
89 stars 126 forks source link

[FEATURE] Add classification algorithms #304

Open ylwu-amzn opened 2 years ago

ylwu-amzn commented 2 years ago

Classification is a common ML requirement. We don't support classification algorithms now. Tribuo supports classification, we can add classification support by integrating with Tribuo. Or other options like a better library supporting classification?

asfoorial commented 1 year ago

Apache OpenNLP is now integrated with HuggingFace Transformers. Could also be an option. In the below example uses a pretrained modelin HuggingFace for text classification.

https://blogs.apache.org/opennlp/entry/accelerate-hugging-face-transformer-models