target code refactor
Browse files
utils/adapmit_classifier.py
CHANGED
|
@@ -35,7 +35,7 @@ def load_adapmitClassifier(config_file:str = None, classifier_name:str = None):
|
|
| 35 |
doc_classifier = pipeline("text-classification",
|
| 36 |
model=classifier_name,
|
| 37 |
return_all_scores=True,
|
| 38 |
-
function_to_apply= "sigmoid")
|
| 39 |
return doc_classifier
|
| 40 |
|
| 41 |
|
|
|
|
| 35 |
doc_classifier = pipeline("text-classification",
|
| 36 |
model=classifier_name,
|
| 37 |
return_all_scores=True,
|
| 38 |
+
function_to_apply= "sigmoid")
|
| 39 |
return doc_classifier
|
| 40 |
|
| 41 |
|