leavoigt commited on
Commit
2312d16
1 Parent(s): b40422c

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")