leavoigt commited on
Commit
513e282
1 Parent(s): 13beabf

Update utils/vulnerability_classifier.py

Browse files
Files changed (1) hide show
  1. utils/vulnerability_classifier.py +6 -0
utils/vulnerability_classifier.py CHANGED
@@ -29,6 +29,12 @@ label_dict= {0: 'Agricultural communities',
29
  17: 'Women and other genders'}
30
 
31
  def getlabels(preds):
 
 
 
 
 
 
32
  # Get label names
33
  preds_list = preds.tolist()
34
 
 
29
  17: 'Women and other genders'}
30
 
31
  def getlabels(preds):
32
+
33
+ """
34
+ Function that takes the numerical predictions as an input and returns a list of the labels.
35
+
36
+ """
37
+
38
  # Get label names
39
  preds_list = preds.tolist()
40