Minh Q. Le commited on
Commit
c53d46b
·
1 Parent(s): a446b0b

Pushed COMIS best model

Browse files
Model/COSMIC/erc_training/epik/README.md ADDED
@@ -0,0 +1 @@
 
 
1
+ Keep feature files here.
Model/COSMIC/erc_training/epik/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c39273f324d37dd49a760d4d99ea59220b3aa7043c296e11f50aa19c03424b2
3
+ size 47576354
Model/COSMIC/feature_extraction/checkpoints/epik/README.md ADDED
@@ -0,0 +1 @@
 
 
1
+ # Model Checkpoints trained on EPIK's dataset
Model/COSMIC/feature_extraction/comet_feature_extract_all.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pickle, numpy as np
2
+ from comet.csk_feature_extract import CSKFeatureExtractor
3
+
4
+ extractor = CSKFeatureExtractor()
5
+
6
+ # ['iemocap', 'dailydialog', 'meld', 'emorynlp']
7
+ for dataset in ['epik']:
8
+ print ('Extracting features in', dataset)
9
+ sentences = pickle.load(open(dataset + '/' + dataset + '_sentences.pkl', 'rb'))
10
+ feaures = extractor.extract(sentences)
11
+ pickle.dump(feaures, open(dataset + '/' + dataset + '_features_comet.pkl', 'wb'))
12
+
13
+ print ('Done!')
Model/COSMIC/feature_extraction/dict.txt ADDED
The diff for this file is too large to render. See raw diff
 
Model/COSMIC/feature_extraction/encoder.json ADDED
The diff for this file is too large to render. See raw diff
 
Model/COSMIC/feature_extraction/requirements.txt ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ absl-py==2.0.0
2
+ annotated-types==0.6.0
3
+ antlr4-python3-runtime==4.8
4
+ asttokens==2.4.1
5
+ bitarray==2.8.2
6
+ blis==0.7.11
7
+ cachetools==5.3.2
8
+ catalogue==2.0.10
9
+ certifi==2023.7.22
10
+ cffi==1.16.0
11
+ charset-normalizer==3.3.1
12
+ click==8.1.7
13
+ cloudpathlib==0.16.0
14
+ colorama==0.4.6
15
+ confection==0.1.3
16
+ cymem==2.0.8
17
+ Cython==3.0.5
18
+ decorator==5.1.1
19
+ en-core-web-sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.0/en_core_web_sm-3.7.0-py3-none-any.whl#sha256=6215d71a3212690e9aec49408a27e3fe6ad7cd6c715476e93d70dc784041e93e
20
+ exceptiongroup==1.1.3
21
+ executing==2.0.1
22
+ fairseq==0.12.2
23
+ filelock==3.12.4
24
+ fsspec==2023.10.0
25
+ ftfy==6.1.1
26
+ google-auth==2.23.3
27
+ google-auth-oauthlib==1.1.0
28
+ grpcio==1.59.0
29
+ hydra-core==1.0.7
30
+ idna==3.4
31
+ ipython==8.17.2
32
+ jedi==0.19.1
33
+ Jinja2==3.1.2
34
+ joblib==1.3.2
35
+ langcodes==3.3.0
36
+ lxml==4.9.3
37
+ Markdown==3.5
38
+ MarkupSafe==2.1.3
39
+ matplotlib-inline==0.1.6
40
+ mpmath==1.3.0
41
+ murmurhash==1.0.10
42
+ networkx==3.2
43
+ nltk==3.8.1
44
+ numpy==1.26.1
45
+ nvidia-cublas-cu12==12.1.3.1
46
+ nvidia-cuda-cupti-cu12==12.1.105
47
+ nvidia-cuda-nvrtc-cu12==12.1.105
48
+ nvidia-cuda-runtime-cu12==12.1.105
49
+ nvidia-cudnn-cu12==8.9.2.26
50
+ nvidia-cufft-cu12==11.0.2.54
51
+ nvidia-curand-cu12==10.3.2.106
52
+ nvidia-cusolver-cu12==11.4.5.107
53
+ nvidia-cusparse-cu12==12.1.0.106
54
+ nvidia-nccl-cu12==2.18.1
55
+ nvidia-nvjitlink-cu12==12.3.52
56
+ nvidia-nvtx-cu12==12.1.105
57
+ oauthlib==3.2.2
58
+ omegaconf==2.0.6
59
+ packaging==23.2
60
+ pandas==2.1.1
61
+ parso==0.8.3
62
+ pexpect==4.8.0
63
+ portalocker==2.8.2
64
+ preshed==3.0.9
65
+ prompt-toolkit==3.0.39
66
+ protobuf==4.23.4
67
+ ptyprocess==0.7.0
68
+ pure-eval==0.2.2
69
+ pyasn1==0.5.0
70
+ pyasn1-modules==0.3.0
71
+ pycparser==2.21
72
+ pydantic==2.4.2
73
+ pydantic_core==2.10.1
74
+ Pygments==2.16.1
75
+ python-dateutil==2.8.2
76
+ pytz==2023.3.post1
77
+ PyYAML==6.0.1
78
+ regex==2023.10.3
79
+ requests==2.31.0
80
+ requests-oauthlib==1.3.1
81
+ rsa==4.9
82
+ sacrebleu==2.3.2
83
+ scikit-learn==1.3.2
84
+ scipy==1.11.3
85
+ six==1.16.0
86
+ smart-open==6.4.0
87
+ spacy==3.7.2
88
+ spacy-legacy==3.0.12
89
+ spacy-loggers==1.0.5
90
+ srsly==2.4.8
91
+ stack-data==0.6.3
92
+ sympy==1.12
93
+ tabulate==0.9.0
94
+ tensorboard==2.15.0
95
+ tensorboard-data-server==0.7.2
96
+ tensorboardX==2.6.2.2
97
+ thinc==8.2.1
98
+ threadpoolctl==3.2.0
99
+ torch==2.1.0
100
+ torch-tb-profiler==0.4.3
101
+ torchaudio==2.1.0
102
+ tqdm==4.66.1
103
+ traitlets==5.13.0
104
+ triton==2.1.0
105
+ typer==0.9.0
106
+ typing_extensions==4.8.0
107
+ tzdata==2023.3
108
+ urllib3==2.0.7
109
+ wasabi==1.1.2
110
+ wcwidth==0.2.9
111
+ weasel==0.3.4
112
+ Werkzeug==3.0.1
Model/COSMIC/feature_extraction/vocab.bpe ADDED
The diff for this file is too large to render. See raw diff