from spiff import TextClassifier prep_documents = [preprocess(document) for document in documents] classifier = TextClassifier() classifier.train(prep_documents, labels) preprocessed_doc = preprocess(test_document) prediction = classifier.predict([preprocessed_doc]) from spiff import FeatureExtractor, TextClassifier feature_extractor = FeatureExtractor() features = feature_extractor.extract(emails) classifier = TextClassifier() classifier.train(features, labels) test_features = feature_extractor.extract([test_email]) prediction = classifier.predict(test_features)