|
|
|
from sklearn.feature_extraction.text import TfidfVectorizer |
|
from sklearn.metrics.pairwise import cosine_similarity |
|
from sklearn.feature_extraction import _stop_words |
|
from convert import ExtractPDFText |
|
|
|
def calculateATSscore(resume_data, job_description): |
|
stopwords = list(_stop_words.ENGLISH_STOP_WORDS) |
|
vectorizer = TfidfVectorizer(stop_words=stopwords) |
|
vectors = vectorizer.fit_transform([job_description, resume_data]) |
|
similarity_value = cosine_similarity(vectors) |
|
print(similarity_value) |
|
|
|
return similarity_value[0,1] |
|
|
|
|
|
|
|
|