Hugging Face's logo

Spaces:
panduwana
/
extractor
Sleeping

App Files Community
mrfirdauss commited on
Commit
7e70df5
·
verified ·
1 Parent(s): cd14bf0

Update classificator.py

Browse files
Files changed (1) hide show
  1. classificator.py +1 -1
classificator.py CHANGED
@@ -48,5 +48,5 @@ def predict(cv, job):
48
  X = pd.DataFrame.from_dict(data)
49
  res = model.predict(X)
50
  results['score'] = model.predict_proba(X)[:, 1]
51
- results['is_accepted'] = np.argmax(res)
52
  return results
 
48
  X = pd.DataFrame.from_dict(data)
49
  res = model.predict(X)
50
  results['score'] = model.predict_proba(X)[:, 1]
51
+ results['is_accepted'] = res[0]
52
  return results