aylajadu/qa-text-classifier

QA Text Classifier

A small question-answering text classifier: TF-IDF features + Logistic Regression, trained on 10 question/answer pairs (AI, ML, deep learning, Python, CPU, GPU, NLP, data science, optimizer, gradient descent).

Intended for demos and teaching of classic ML pipelines.

Usage

import pickle

with open("qa_model.pkl", "rb") as f:
    bundle = pickle.load(f)

q = "What is AI?"
print(bundle["classifier"].predict(bundle["vectorizer"].transform([q]))[0])

License

apache-2.0