ashuonnet commited on
Commit
786353d
·
1 Parent(s): b0901c5

Update skill_recommender.py

Browse files
Files changed (1) hide show
  1. skill_recommender.py +11 -1
skill_recommender.py CHANGED
@@ -63,7 +63,17 @@ class skills_recommender:
63
  def get_vertical_name(self, vertical):
64
  return self.vertical_mapping[vertical]
65
 
66
- def suggest_skills(self,skill_list):
 
 
 
 
 
 
 
 
 
 
67
  print("checking for skill list = ",skill_list)
68
  df = self.get_matching_vertical_skills(skill_list)
69
  df['vertical'].value_counts()
 
63
  def get_vertical_name(self, vertical):
64
  return self.vertical_mapping[vertical]
65
 
66
+ def suggest_skills(self,org_skill_list):
67
+ #clean skill list if needed
68
+ skill_list = []
69
+ for _skill in org_skill_list:
70
+ if "\n" in _skill:
71
+ _skill = _skill.replace("\n",",")
72
+ skill_list = skill_list + _skill.split(',')
73
+ else:
74
+ skill_list.append(_skill)
75
+
76
+
77
  print("checking for skill list = ",skill_list)
78
  df = self.get_matching_vertical_skills(skill_list)
79
  df['vertical'].value_counts()