File size: 290 Bytes
88b8fd6 |
1 2 3 4 5 |
from binary_classifier import CBTBinaryClassifier
classifier = CBTBinaryClassifier()
classifier.load_model('./cbt_classifier')
result = classifier.predict('I am happy cause I finished all of my tasks')
print(f"Prediction: {result['is_cbt_trigger']}, Confidence: {result['confidence']:.3f}") |