Spaces:
Runtime error
Runtime error
Create ATS_score.py
Browse files- ATS_score.py +14 -0
ATS_score.py
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#find ats score
|
2 |
+
from sklearn.feature_extraction.text import TfidfVectorizer
|
3 |
+
from sklearn.metrics.pairwise import cosine_similarity
|
4 |
+
from sklearn.feature_extraction import _stop_words
|
5 |
+
from convert import ExtractPDFText
|
6 |
+
|
7 |
+
def calculateATSscore(resume_data, job_description):
|
8 |
+
stopwords = list(_stop_words.ENGLISH_STOP_WORDS)
|
9 |
+
vectorizer = TfidfVectorizer(stop_words=stopwords)
|
10 |
+
vectors = vectorizer.fit_transform([job_description, resume_data])
|
11 |
+
similarity_value = cosine_similarity(vectors)
|
12 |
+
print(similarity_value)
|
13 |
+
# return similarity_value[0,1]
|
14 |
+
return similarity_value[0,1]
|