prashant
commited on
Commit
•
caf492f
1
Parent(s):
fc140bc
float error
Browse files- utils/sdg_classifier.py +1 -1
utils/sdg_classifier.py
CHANGED
@@ -49,7 +49,7 @@ def load_sdgClassifier(configFile = None, docClassifierModel = None):
|
|
49 |
|
50 |
@st.cache(allow_output_mutation=True)
|
51 |
def sdg_classification(haystackdoc:List[Document],
|
52 |
-
threshold:
|
53 |
"""
|
54 |
Text-Classification on the list of texts provided. Classifier provides the
|
55 |
most appropriate label for each text. these labels are in terms of if text
|
|
|
49 |
|
50 |
@st.cache(allow_output_mutation=True)
|
51 |
def sdg_classification(haystackdoc:List[Document],
|
52 |
+
threshold:float, classifiermodel)->Tuple[DataFrame,Series]:
|
53 |
"""
|
54 |
Text-Classification on the list of texts provided. Classifier provides the
|
55 |
most appropriate label for each text. these labels are in terms of if text
|