Spaces:
Running
Running
update touchpoint model
Browse files- .ipynb_checkpoints/utils-checkpoint.py +1 -1
- 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': ['
|
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': ['
|
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 |
|