Added a base model class and training scripts for various sentiment analysis models, including Naive Bayes, SVM, XGBoost, LSTM, and BERT. Also, improved prediction functionality and the model loading mechanism.
This commit is contained in:
@@ -1,6 +1,9 @@
|
||||
jieba==0.42.1
|
||||
pytorch==1.7.1
|
||||
sklearn==0.23.2
|
||||
xgboost==1.2.1
|
||||
transformers==4.6.1
|
||||
gensim==4.0.1
|
||||
torch>=1.7.1
|
||||
scikit-learn>=0.23.2
|
||||
xgboost>=1.2.1
|
||||
transformers>=4.6.1
|
||||
gensim>=4.0.1
|
||||
pandas>=1.3.0
|
||||
numpy>=1.20.0
|
||||
tqdm>=4.60.0
|
||||
Reference in New Issue
Block a user