mrfirdauss commited on
Commit
34571d7
·
verified ·
1 Parent(s): 99917de

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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.predict(X)[:, 1]
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