Upload folder using huggingface_hub
Browse files- classificator.py +1 -1
classificator.py
CHANGED
@@ -35,7 +35,7 @@ def predict(cv, job):
|
|
35 |
role_pos = cosine_similarity(st.encode(cv['positions']).reshape(1,-1), st.encode(job['role']).reshape(1,-1))[0][0]
|
36 |
major_similarity = cosine_similarity(st.encode(cv['userMajors']).reshape(1,-1), st.encode(job['majors']).reshape(1,-1))[0][0]
|
37 |
skill_similarity = cosine_similarity(st.encode(cv['skills']).reshape(1,-1), st.encode(job['skills']).reshape(1,-1))[0][0]
|
38 |
-
score_yoe = 0.5 if diffYoe == -1 else (
|
39 |
score = 0.35 * role_req_exp + 0.1 * role_pos + 0.15 * major_similarity + 0.3* score_yoe + 0.1 * skill_similarity
|
40 |
X = np.array([role_req_exp, role_pos, major_similarity, skill_similarity, score]).reshape(1, -1)
|
41 |
res = model.predict(X)
|
|
|
35 |
role_pos = cosine_similarity(st.encode(cv['positions']).reshape(1,-1), st.encode(job['role']).reshape(1,-1))[0][0]
|
36 |
major_similarity = cosine_similarity(st.encode(cv['userMajors']).reshape(1,-1), st.encode(job['majors']).reshape(1,-1))[0][0]
|
37 |
skill_similarity = cosine_similarity(st.encode(cv['skills']).reshape(1,-1), st.encode(job['skills']).reshape(1,-1))[0][0]
|
38 |
+
score_yoe = 0.5 if diffYoe == -1 else (0 if diffYoe < 0 else 1)
|
39 |
score = 0.35 * role_req_exp + 0.1 * role_pos + 0.15 * major_similarity + 0.3* score_yoe + 0.1 * skill_similarity
|
40 |
X = np.array([role_req_exp, role_pos, major_similarity, skill_similarity, score]).reshape(1, -1)
|
41 |
res = model.predict(X)
|