Rename appStore/identifier.py to appStore/target.py
Browse files
appStore/{identifier.py → target.py}
RENAMED
@@ -15,5 +15,12 @@ from utils.config import get_classifier_params
|
|
15 |
classifier_identifier = 'target'
|
16 |
params = get_classifier_params(classifier_identifier)
|
17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
# Download model from HF Hub
|
19 |
vg_model = SetFitModel.from_pretrained("leavoigt/vulnerable_groups")
|
|
|
15 |
classifier_identifier = 'target'
|
16 |
params = get_classifier_params(classifier_identifier)
|
17 |
|
18 |
+
## Labels dictionary ###
|
19 |
+
_lab_dict = {
|
20 |
+
'NEGATIVE':'NO TARGET INFO',
|
21 |
+
'TARGET':'TARGET',
|
22 |
+
}
|
23 |
+
|
24 |
+
|
25 |
# Download model from HF Hub
|
26 |
vg_model = SetFitModel.from_pretrained("leavoigt/vulnerable_groups")
|