https://github.com/anurima-saha/text-classification

Wine reviews used to determine the type of wine training on imbalanced data using classification algorithms like SVM, Naive Bayes and Random Forest Classifier. Neural Network (CNN, RNN and LSTM) and LLM models (DistilBERT and RoBERTa) were also used followed by error analysis using SHAP.

distill-bert llm llm-finetuning naive-bayes-classifier random-forest-classifier rnn-pytorch roberta-model shapley-value svm-classifier text-classification transfer-learning

Last synced: 26 Feb 2025