Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- classificator.py +1 -1
classificator.py
CHANGED
@@ -39,6 +39,6 @@ def predict(cv, job):
|
|
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)
|
42 |
-
results['score'] = model.
|
43 |
results['is_accepted'] = np.argmax(res)
|
44 |
return results
|
|
|
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)
|
42 |
+
results['score'] = model.predict_proba(X)[:, 1]
|
43 |
results['is_accepted'] = np.argmax(res)
|
44 |
return results
|