Binary_classifier / test_model.py
SaitejaJate's picture
Upload 5 files
88b8fd6 verified
raw
history blame contribute delete
290 Bytes
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}")