Hyma7 commited on
Commit
bb6e8b2
·
verified ·
1 Parent(s): 7f54a66

Update skill_matcher.py

Browse files
Files changed (1) hide show
  1. skill_matcher.py +20 -20
skill_matcher.py CHANGED
@@ -1,20 +1,20 @@
1
- from sklearn.feature_extraction.text import TfidfVectorizer
2
- from sklearn.metrics.pairwise import cosine_similarity
3
- import json
4
-
5
- def load_job_descriptions(path='data/job_descriptions.json'):
6
- with open(path, 'r') as f:
7
- jobs = json.load(f)
8
- return jobs
9
-
10
- def compare_with_jobs(resume_text, job_descriptions):
11
- results = []
12
- vectorizer = TfidfVectorizer().fit_transform([resume_text] + [job["description"] for job in job_descriptions])
13
- vectors = vectorizer.toarray()
14
-
15
- resume_vector = vectors[0]
16
- for idx, job in enumerate(job_descriptions):
17
- job_vector = vectors[idx + 1]
18
- similarity = cosine_similarity([resume_vector], [job_vector])[0][0]
19
- results.append({"role": job["role"], "match": round(similarity * 100, 2)})
20
- return sorted(results, key=lambda x: x["match"], reverse=True)
 
1
+ from sklearn.feature_extraction.text import TfidfVectorizer
2
+ from sklearn.metrics.pairwise import cosine_similarity
3
+ import json
4
+
5
+ def load_job_descriptions(path='job_descriptions.json'):
6
+ with open(path, 'r') as f:
7
+ jobs = json.load(f)
8
+ return jobs
9
+
10
+ def compare_with_jobs(resume_text, job_descriptions):
11
+ results = []
12
+ vectorizer = TfidfVectorizer().fit_transform([resume_text] + [job["description"] for job in job_descriptions])
13
+ vectors = vectorizer.toarray()
14
+
15
+ resume_vector = vectors[0]
16
+ for idx, job in enumerate(job_descriptions):
17
+ job_vector = vectors[idx + 1]
18
+ similarity = cosine_similarity([resume_vector], [job_vector])[0][0]
19
+ results.append({"role": job["role"], "match": round(similarity * 100, 2)})
20
+ return sorted(results, key=lambda x: x["match"], reverse=True)