Hugging Face's logo

Spaces:
hongaik
/
hc_text_classification
Sleeping

App Files Community
hongaik commited on
Commit
e22f095
•
1 Parent(s): 4d9c486

update touchpoint model

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/utils-checkpoint.py +1 -1
  2. utils.py +1 -1
.ipynb_checkpoints/utils-checkpoint.py CHANGED
@@ -39,7 +39,7 @@ def get_single_prediction(text):
39
  # Make touchpoint predictions
40
  text_vectors_touchpoint = tfidf_touchpoint.transform([text])
41
  results_touchpoint = svc_touchpoint.predict_proba(text_vectors_touchpoint).squeeze().round(2)
42
- pred_prob_touchpoint = pd.DataFrame({'touchpoint': ['No touchpoint', 'ASC', 'CC', 'Technician'], 'probability': results_touchpoint}).sort_values('probability', ascending=True)
43
 
44
  return (pred_prob, pred_prob_sentiment, pred_prob_touchpoint)
45
 
 
39
  # Make touchpoint predictions
40
  text_vectors_touchpoint = tfidf_touchpoint.transform([text])
41
  results_touchpoint = svc_touchpoint.predict_proba(text_vectors_touchpoint).squeeze().round(2)
42
+ pred_prob_touchpoint = pd.DataFrame({'touchpoint': ['ASC', 'CC', 'No touchpoint', 'Technician'], 'probability': results_touchpoint}).sort_values('probability', ascending=True)
43
 
44
  return (pred_prob, pred_prob_sentiment, pred_prob_touchpoint)
45
 
utils.py CHANGED
@@ -39,7 +39,7 @@ def get_single_prediction(text):
39
  # Make touchpoint predictions
40
  text_vectors_touchpoint = tfidf_touchpoint.transform([text])
41
  results_touchpoint = svc_touchpoint.predict_proba(text_vectors_touchpoint).squeeze().round(2)
42
- pred_prob_touchpoint = pd.DataFrame({'touchpoint': ['No touchpoint', 'ASC', 'CC', 'Technician'], 'probability': results_touchpoint}).sort_values('probability', ascending=True)
43
 
44
  return (pred_prob, pred_prob_sentiment, pred_prob_touchpoint)
45
 
 
39
  # Make touchpoint predictions
40
  text_vectors_touchpoint = tfidf_touchpoint.transform([text])
41
  results_touchpoint = svc_touchpoint.predict_proba(text_vectors_touchpoint).squeeze().round(2)
42
+ pred_prob_touchpoint = pd.DataFrame({'touchpoint': ['ASC', 'CC', 'No touchpoint', 'Technician'], 'probability': results_touchpoint}).sort_values('probability', ascending=True)
43
 
44
  return (pred_prob, pred_prob_sentiment, pred_prob_touchpoint)
45