From: Exploring best-matched embedding model and classifier for charging-pile fault diagnosis
Environment | Libraries |
---|---|
Python 3.7.13 | Pandas 1.3.5 |
Numpy 1.21.5 Jieba 0.42.1 Gensim 4.1.2 Imbalance-learn 0.9.0 | |
Scikit-learn 1.0.2 Pydotplus 2.0.2 Pytorch 1.12.0 Cudatoolkit 11.3.1 Transformers 4.20.1 | |
GloVe 1.2 | – |
BERT (pre-trained) | Chinese_wwm_ext_pytorch |
Word2vector | Gensim.models.word2vec (in gensim 4.1.2) |