-
Notifications
You must be signed in to change notification settings - Fork 5
/
environment.yml
37 lines (36 loc) · 1.31 KB
/
environment.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
name: lady
channels:
- conda-forge
- pytorch
- nvidia
- anaconda
- defaults
dependencies:
- pip
- python=3.8
- pandas==1.4.4
- seaborn
- spacy==3.7.2
#spacy-transformers
#python -m spacy download en_core_web_trf
#python -m spacy download en_core_web_sm
- tqdm
- transformers==4.26.1 #needed if augmentation via backtranslation
- pytorch==1.13.1
- pytorch-cuda=11.8 # the conda channel has issue, so pip3 install torch --index-url https://download.pytorch.org/whl/cu118
- scikit-learn>=1.0.0
- gensim==4.*
- nltk==3.7
- pampy==0.3.0
- returns==0.22.0
- more-itertools==10.1.0
# - matplotlib
- pip:
- sentence-transformers #for semantic similarity of backtranslated review with its original review
- simalign==0.3 #for aligning words of backtranslated review to the original review
- rouge #for evaluating backtranlated reviews vs. original review
- bitermplus==0.6.10 #install it if you want to use btm model in the aspect modeling layer
- contextualized_topic_models #install it if you want to use ctm model in the aspect modeling layer
- natsort
- pytrec-eval-terrier
# - octis #use Fani's Lab! fork at https://github.com/fani-lab/OCTIS followed by "cd src/octis/; python setup.py install"