File size: 836 Bytes
be57e68 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
"""
Simple test script for the text classification model
"""
from text_classifier import TextClassifier
def test_model():
classifier = TextClassifier()
# Test cases
test_cases = [
("I love this product!", "positive"),
("This is terrible.", "negative"),
("It's okay, nothing special.", None), # Could be positive or neutral
]
print("Testing Text Classification Model:")
print("-" * 40)
for text, expected_label in test_cases:
result = classifier.predict(text)
print(f"Input: {text}")
print(f"Predicted: {result['label']} (confidence: {result['confidence']})")
if expected_label:
print(f"Expected: {expected_label}")
print()
print("Test completed successfully!")
if __name__ == "__main__":
test_model() |