netzero check
Browse files
utils/netzero_classifier.py
CHANGED
|
@@ -73,7 +73,9 @@ def netzero_classification(haystack_doc:pd.DataFrame,
|
|
| 73 |
haystack_doc['Netzero Label'] = 'NA'
|
| 74 |
haystack_doc['Netzero Score'] = 'NA'
|
| 75 |
temp = haystack_doc[haystack_doc['Target Label'] == 'TARGET']
|
|
|
|
| 76 |
df = haystack_doc[haystack_doc['Target Label'] == 'NEGATIVE']
|
|
|
|
| 77 |
|
| 78 |
if not classifier_model:
|
| 79 |
classifier_model = st.session_state['netzero_classifier']
|
|
|
|
| 73 |
haystack_doc['Netzero Label'] = 'NA'
|
| 74 |
haystack_doc['Netzero Score'] = 'NA'
|
| 75 |
temp = haystack_doc[haystack_doc['Target Label'] == 'TARGET']
|
| 76 |
+
temp = temp.reset_index(drop=True)
|
| 77 |
df = haystack_doc[haystack_doc['Target Label'] == 'NEGATIVE']
|
| 78 |
+
df = df.reset_index(drop=True)
|
| 79 |
|
| 80 |
if not classifier_model:
|
| 81 |
classifier_model = st.session_state['netzero_classifier']
|