environment
python3.8
store datasets
- .vector_cache # word vector ,auto download to the .vector_cache
- .data # if selected imdb, auto download dataset to the directory.
Depends on
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
python3 -m spacy download en_core_web_sm
Datasets download
链接：https://pan.baidu.com/s/1XCmt5xlAE0yKcYWLKpFQqQ 提取码：GZZ0
example
# step 1
# create FasterSentiment object
# abs_dataset_path your datasets dir
sobj = FasterSent(abs_dataset_path=root_path, epochs=20)
# step 2
# start training
sobj.start_train()
# step 3
# evaluate model
res = sobj.start_eval('tut3-model.pt')
print(res)
# step 3
# predict text
res = sobj.predict("He is a bad bad boy!")
print(res)
